Add a script to test dqn variants all at once

keiohta · May 29, 2019 · 5b455ec · 5b455ec
1 parent 545af35
commit 5b455ec
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/examples/run_all_dqn.sh b/examples/run_all_dqn.sh
@@ -0,0 +1,14 @@
+rm -rf results
+python examples/run_dqn.py --dir-suffix=dqn
+# DDQN
+python examples/run_dqn.py --enable-double-dqn --dir-suffix=ddqn
+# Dueling DDQN
+python examples/run_dqn.py --enable-double-dqn --enable-dueling-dqn --dir-suffix=dueling_ddqn
+# Noisy Nets
+python examples/run_dqn.py --enable-noisy-dqn --dir-suffix=noisy
+# Categorical DQN
+python examples/run_dqn.py --enable-categorical-dqn --dir-suffix=categorical_dqn
+# Dueling DDQN Noisy Nets
+python examples/run_dqn.py --enable-double-dqn --enable-dueling-dqn --enable-noisy-dqn --dir-suffix=dueling_ddqn_noisy
+# Dueling DDQN Categorical Noisy Nets
+python examples/run_dqn.py --enable-double-dqn --enable-dueling-dqn --enable-noisy-dqn --enable-categorical-dqn --dir-suffix=dueling_ddqn_categorical_noisy
diff --git a/examples/run_dqn.py b/examples/run_dqn.py
@@ -9,6 +9,7 @@
     parser = DQN.get_argument(parser)
     parser.set_defaults(test_interval=2000)
     parser.set_defaults(max_steps=int(5e5))
+    parser.set_defaults(max_steps=100000)
     parser.set_defaults(gpu=-1)
     args = parser.parse_args()