Update run_simulation.py

yjwong1999 · Jun 14, 2023 · 8607560 · 8607560
1 parent 69853c3
commit 8607560
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/run_simulation.py b/run_simulation.py
@@ -6,10 +6,10 @@
 
 # get argument from user
 parser = argparse.ArgumentParser()
-parser.add_argument('--store-path', type = str, required = True, help="pretrained model weight path")
+parser.add_argument('--path', type = str, required = True, help="pretrained model weight path")
 
 args = parser.parse_args()
-STORE_PATH = args.store_path
+STORE_PATH = args.path
 
 # validate the weight path
 if not os.path.isdir(STORE_PATH):