"training every 5 steps for 100episodes"

scioip34 · May 20, 2024 · bd68303 · bd68303
1 parent dacf97e
commit bd68303
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 8 deletions.
diff --git a/abm/data/metaprotocol/experiments/MADRLExp0N3R100T5M6.py b/abm/data/metaprotocol/experiments/MADRLExp0N3R100T5M6.py
@@ -27,7 +27,7 @@
     Constant("ENV_WIDTH", 500),
     Constant("ENV_HEIGHT", 500),
     Constant("WINDOW_PAD", 30),
-    Constant("N_EPISODES", 50),
+    Constant("N_EPISODES", 100),
     Constant("T", 20000),
     Constant("SEED", 0),
     Constant("TRAIN", 1),
@@ -41,7 +41,7 @@
     Constant("EPSILON_END", 0.001),
     Constant("EPSILON_DECAY", 50000),
     Constant("TAU", 0.01),
-    Constant("OPTIMIZER", "RMSPROP"),
+    Constant("OPTIMIZER", "ADAM"),
     Constant("PRETRAINED_MODELS_DIR", ""),
     Constant("BRAIN_TYPE", "DQN"),
     #Constant("ISE_W", 1.0),

diff --git a/abm/data/metaprotocol/experiments/MADRLExp0N3R10T5M6.py b/abm/data/metaprotocol/experiments/MADRLExp0N3R10T5M6.py
@@ -27,7 +27,7 @@
     Constant("ENV_WIDTH", 500),
     Constant("ENV_HEIGHT", 500),
     Constant("WINDOW_PAD", 30),
-    Constant("N_EPISODES", 50),
+    Constant("N_EPISODES", 100),
     Constant("T", 20000),
     Constant("SEED", 0),
     Constant("TRAIN", 1),
@@ -41,7 +41,7 @@
     Constant("EPSILON_END", 0.001),
     Constant("EPSILON_DECAY", 50000),
     Constant("TAU", 0.01),
-    Constant("OPTIMIZER", "RMSPROP"),
+    Constant("OPTIMIZER", "ADAM"),
     Constant("PRETRAINED_MODELS_DIR", ""),
     Constant("BRAIN_TYPE", "DQN"),
     #Constant("ISE_W", 1.0),

diff --git a/abm/data/metaprotocol/experiments/MADRLExp0N3R3T5M6.py b/abm/data/metaprotocol/experiments/MADRLExp0N3R3T5M6.py
@@ -25,7 +25,7 @@
     Constant("ENV_WIDTH", 500),
     Constant("ENV_HEIGHT", 500),
     Constant("WINDOW_PAD", 30),
-    Constant("N_EPISODES", 50),
+    Constant("N_EPISODES", 100),
     Constant("T", 20000),
     Constant("SEED", 0),
     Constant("TRAIN", 1),
@@ -39,7 +39,7 @@
     Constant("EPSILON_END", 0.01),
     Constant("EPSILON_DECAY", 50000),
     Constant("TAU", 0.01),
-    Constant("OPTIMIZER", "RMSPROP"),
+    Constant("OPTIMIZER", "ADAM"),
     Constant("PRETRAINED_MODELS_DIR", ""),
     Constant("BRAIN_TYPE", "DQN"),
     #Constant("ISE_W", 1.0),

diff --git a/abm/data/metaprotocol/experiments/MADRLExp0N3R50T5M6.py b/abm/data/metaprotocol/experiments/MADRLExp0N3R50T5M6.py
@@ -27,7 +27,7 @@
     Constant("ENV_WIDTH", 500),
     Constant("ENV_HEIGHT", 500),
     Constant("WINDOW_PAD", 30),
-    Constant("N_EPISODES", 50),
+    Constant("N_EPISODES", 100),
     Constant("T", 20000),
     Constant("SEED", 0),
     Constant("TRAIN", 1),
@@ -41,7 +41,7 @@
     Constant("EPSILON_END", 0.001),
     Constant("EPSILON_DECAY", 50000),
     Constant("TAU", 0.01),
-    Constant("OPTIMIZER", "RMSPROP"),
+    Constant("OPTIMIZER", "ADAM"),
     Constant("PRETRAINED_MODELS_DIR", ""),
     Constant("BRAIN_TYPE", "DQN"),
     #Constant("ISE_W", 1.0),