remove unneeded args.

gsbDBI · Jun 27, 2023 · eb1dd51 · eb1dd51
1 parent 42b9200
commit eb1dd51
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/bemb/utils/run_helper_lightning.py b/bemb/utils/run_helper_lightning.py
@@ -23,9 +23,9 @@ def run(model: "LitBEMBFlex",
         dataset_train: ChoiceDataset,
         dataset_val: Optional[ChoiceDataset]=None,
         dataset_test: Optional[ChoiceDataset]=None,
-        model_optimizer: str='Adam',
+        # model_optimizer: str='Adam',
         batch_size: int=-1,
-        learning_rate: float=0.01,
+        # learning_rate: float=0.01,
         num_epochs: int=10,
         num_workers: int=0,
         device: Optional[str]=None,