Update run_simulation.py

yjwong1999 · Jun 14, 2023 · 2583e09 · 2583e09
1 parent 2e38a84
commit 2583e09
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/run_simulation.py b/run_simulation.py
@@ -7,11 +7,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('--ep-num', type = int, required = True, default=300, help='total number of episodes')
 
 args = parser.parse_args()
 STORE_PATH = args.store_path
-EP_NUM = args.ep_num
+EP_NUM = 1
 
 # validate the weight path
 if not os.path.isdir(STORE_PATH):