Fix typo

torchmd · Apr 17, 2024 · f54adb3 · f54adb3
1 parent 160e3c4
commit f54adb3
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/torchmdnet/scripts/train.py b/torchmdnet/scripts/train.py
@@ -121,9 +121,9 @@ def get_argparse():
     parser.add_argument('--wandb-project', default='training_', type=str, help='Define what wandb Project to log to')
     parser.add_argument('--wandb-resume-from-id', default=None, type=str, help='Resume a wandb run from a given run id. The id can be retrieved from the wandb dashboard')
     parser.add_argument('--tensorboard-use', default=False, type=bool, help='Defines if tensor board is used or not')
-    parser.add_argument('--freeze_representation', type=bool, default=False, help='Freeze the representation model parameters during training.')
-    parser.add_argument('--reset_output_model', type=bool, default=False, help='Reset the parameters (randomize) of the output models before stating a training. This option is only used if the training is not starting from scratch, otherwise the parameters are always randomized.')
-    parser.add_argument('--overwrite_representation', type=str, help='After loading/creating the model, overwrite the weights of the representation model using the ones stored in the checkpoint provided in this argument.')
+    parser.add_argument('--freeze-representation', type=bool, default=False, help='Freeze the representation model parameters during training.')
+    parser.add_argument('--reset-output-model', type=bool, default=False, help='Reset the parameters (randomize) of the output models before stating a training. This option is only used if the training is not starting from scratch, otherwise the parameters are always randomized.')
+    parser.add_argument('--overwrite-representation', type=str, help='After loading/creating the model, overwrite the weights of the representation model using the ones stored in the checkpoint provided in this argument.')
     # fmt: on
     return parser