Update run_simulation.py

yjwong1999 · Jun 14, 2023 · 454be0d · 454be0d
1 parent 5372ed5
commit 454be0d
Showing 1 changed file with 11 additions and 4 deletions.
diff --git a/run_simulation.py b/run_simulation.py
@@ -14,14 +14,21 @@
 # validate the weight path
 if not os.path.isdir(STORE_PATH):
  raise NotImplementedError("The provided weight path does not exist!")
+
 # get DRL_ALGO
-if 'ddpg' in STORE_PATH:
+if 'td3' in STORE_PATH:
+ DRL_ALGO = 'td3'
+else:
  DRL_ALGO = 'ddpg'
+
+# reward design
+if 'see' in STORE_PATH:
+ REWARD_DESIGN = 'see'
 else:
- DRL_ALGO = 'td3'
+ REWARD_DESIGN = 'ssr'
 
-REWARD_DESIGN = args.reward
-SEEDS = args.seeds
+# seeds and episode number
+SEEDS = None
 EPISODE_NUM = args.ep_num