Update Readme

filippo-merlo · Jul 7, 2023 · 923a770 · 923a770
1 parent df9f99a
commit 923a770
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -57,6 +57,9 @@ Highlights:
 ### Learning
 To train the models from scratch:
 ```
+python main.py [-h] --in_path IN_PATH --out_path OUT_PATH
+               [--model_name MODEL_NAME] [--pre_train PRE_TRAIN]
+
 optional arguments:
   -h, --help            show this help message and exit
   --in_path IN_PATH, -i IN_PATH