Update run_simulation.py

yjwong1999 · Jun 14, 2023 · b578c78 · b578c78
1 parent 7d84198
commit b578c78
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/run_simulation.py b/run_simulation.py
@@ -10,7 +10,6 @@
 
 args = parser.parse_args()
 STORE_PATH = args.store_path
-EP_NUM = 1
 
 # validate the weight path
 if not os.path.isdir(STORE_PATH):
@@ -50,7 +49,7 @@
 import torch
 
 # 1 init system model
-episode_num = EPISODE_NUM # recommend to be 300
+episode_num = 1
 episode_cnt = 0
 step_num = 100