Update model.py

src/models/model.py

@@ -301,6 +301,7 @@ class Summarization:
             early_stopping_patience_epochs (int, optional): monitors val_loss on epoch end and stops training,
             if val_loss does not improve after the specied number of epochs. set 0 to disable early stopping.
             Defaults to 0 (disabled)
+            :param learning_rate:
         """
         self.target_max_token_len = target_max_token_len
         self.data_module = PLDataModule(