Change training schedule

yura-hb · Apr 24, 2024 · 1212bf4 · 1212bf4
1 parent 07cda6e
commit 1212bf4
Show file tree

Hide file tree

Showing 10 changed files with 35 additions and 35 deletions.
diff --git a/diploma_thesis/configuration/experiments/jsp/GRAPH-NN/experiments/2 (DQN)/experiment.yml b/diploma_thesis/configuration/experiments/jsp/GRAPH-NN/experiments/2 (DQN)/experiment.yml
@@ -315,13 +315,13 @@ task:
 #                  - *marl_dqn_2
                   - *dqn_3
 #                  - *marl_dqn_4
-            - output_dir: 'GAT'
-              machine_agent:
-                parameters:
-                  - *dqn_1_gat
-#                  - *marl_dqn_2_gin
-                  - *dqn_3_gat
-#                  - *marl_dqn_4_gin
+#            - output_dir: 'GAT'
+#              machine_agent:
+#                parameters:
+#                  - *dqn_1_gat
+##                  - *marl_dqn_2_gin
+#                  - *dqn_3_gat
+##                  - *marl_dqn_4_gin
             - output_dir: 'SAGE'
               machine_agent:
                 parameters:

diff --git a/diploma_thesis/configuration/experiments/jsp/GRAPH-NN/experiments/2 (PPO)/experiment.yml b/diploma_thesis/configuration/experiments/jsp/GRAPH-NN/experiments/2 (PPO)/experiment.yml
@@ -198,10 +198,10 @@ task:
               machine_agent:
                 parameters:
                   *ppo_1
-            - output_dir: 'GAT'
-              machine_agent:
-                parameters:
-                  *ppo_2
+#            - output_dir: 'GAT'
+#              machine_agent:
+#                parameters:
+#                  *ppo_2
             - output_dir: 'GCN'
               machine_agent:
                 parameters:

diff --git a/diploma_thesis/configuration/experiments/jsp/GRAPH-NN/experiments/5 (DQN)/experiment.yml b/diploma_thesis/configuration/experiments/jsp/GRAPH-NN/experiments/5 (DQN)/experiment.yml
@@ -307,20 +307,20 @@ task:
                         values:
         values:
           __concat__:
-#            - output_dir: 'Baseline'
-#              machine_agent:
-#                parameters:
-#                  - *dqn_1
-##                  - *marl_dqn_2
-#                  - *dqn_3
-##                  - *marl_dqn_4
-#            - output_dir: 'All'
-#              machine_agent:
-#                parameters:
-#                  - *dqn_1_all
-##                  - *marl_dqn_2_all
-#                  - *dqn_3_all
-##                  - *marl_dqn_4_all
+            - output_dir: 'Baseline'
+              machine_agent:
+                parameters:
+                  - *dqn_1
+#                  - *marl_dqn_2
+                  - *dqn_3
+#                  - *marl_dqn_4
+            - output_dir: 'All'
+              machine_agent:
+                parameters:
+                  - *dqn_1_all
+#                  - *marl_dqn_2_all
+                  - *dqn_3_all
+#                  - *marl_dqn_4_all
             - output_dir: 'Flexible'
               machine_agent:
                 parameters:

diff --git a/diploma_thesis/configuration/experiments/jsp/GRAPH-NN/experiments/5 (PPO)/experiment.yml b/diploma_thesis/configuration/experiments/jsp/GRAPH-NN/experiments/5 (PPO)/experiment.yml
@@ -206,10 +206,10 @@ task:
               machine_agent:
                 parameters:
                   *ppo_3
-            - output_dir: 'Flexible GAT'
-              machine_agent:
-                parameters:
-                  *ppo_4
+#            - output_dir: 'Flexible GAT'
+#              machine_agent:
+#                parameters:
+#                  *ppo_4
 
           tape:
             machine_reward:

diff --git a/diploma_thesis/configuration/experiments/jsp/GRAPH-NN/flexible_machine.yml b/diploma_thesis/configuration/experiments/jsp/GRAPH-NN/flexible_machine.yml
@@ -60,7 +60,7 @@ parameters:
         kind: 'replay'
         parameters:
           size: 2048
-          batch_size: 192
+          batch_size: 256
           prefetch: 8
 
       loss:

diff --git a/diploma_thesis/configuration/experiments/jsp/GRAPH-NN/flexible_marl_machine.yml b/diploma_thesis/configuration/experiments/jsp/GRAPH-NN/flexible_marl_machine.yml
@@ -61,7 +61,7 @@ parameters:
         kind: 'replay'
         parameters:
           size: 2048
-          batch_size: 192
+          batch_size: 256
           prefetch: 8
 
       loss:

diff --git a/diploma_thesis/configuration/experiments/jsp/GRAPH-NN/flexible_ppo_machine.yml b/diploma_thesis/configuration/experiments/jsp/GRAPH-NN/flexible_ppo_machine.yml
@@ -49,7 +49,7 @@ parameters:
     kind: 'ppo'
     parameters:
       device: 'cpu'
-      sample_count: 192
+      sample_count: 256
       policy_step_ratio: 0.2
       entropy_regularization: 0.01
       rollback_ratio: 0.00

diff --git a/diploma_thesis/configuration/experiments/jsp/GRAPH-NN/machine.yml b/diploma_thesis/configuration/experiments/jsp/GRAPH-NN/machine.yml
@@ -60,7 +60,7 @@ parameters:
         kind: 'replay'
         parameters:
           size: 2048
-          batch_size: 192
+          batch_size: 256
           prefetch: 8
 
       loss:

diff --git a/diploma_thesis/configuration/experiments/jsp/GRAPH-NN/marl_machine.yml b/diploma_thesis/configuration/experiments/jsp/GRAPH-NN/marl_machine.yml
@@ -61,7 +61,7 @@ parameters:
         kind: 'replay'
         parameters:
           size: 2048
-          batch_size: 192
+          batch_size: 256
           prefetch: 8
 
       loss:

diff --git a/diploma_thesis/configuration/experiments/jsp/GRAPH-NN/ppo_machine.yml b/diploma_thesis/configuration/experiments/jsp/GRAPH-NN/ppo_machine.yml
@@ -49,7 +49,7 @@ parameters:
     kind: 'ppo'
     parameters:
       device: 'cpu'
-      sample_count: 192
+      sample_count: 256
       policy_step_ratio: 0.2
       entropy_regularization: 0.01
       rollback_ratio: 0.00