Fix training script

yura-hb · Apr 25, 2024 · 051b5e5 · 051b5e5
1 parent ac0769a
commit 051b5e5
Show file tree

Hide file tree

Showing 6 changed files with 10 additions and 10 deletions.
diff --git a/diploma_thesis/configuration/experiments/jsp/GRAPH-NN/flexible_machine.yml b/diploma_thesis/configuration/experiments/jsp/GRAPH-NN/flexible_machine.yml
@@ -57,8 +57,8 @@ parameters:
       memory:
         kind: 'replay'
         parameters:
-          size: 2048
-          batch_size: 256
+          size: 1024
+          batch_size: 128
           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
@@ -60,8 +60,8 @@ parameters:
       memory:
         kind: 'replay'
         parameters:
-          size: 2048
-          batch_size: 256
+          size: 1024
+          batch_size: 128
           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
@@ -48,7 +48,7 @@ parameters:
     kind: 'ppo'
     parameters:
       device: 'cpu'
-      sample_count: 256
+      sample_count: 128
       policy_step_ratio: 0.2
       entropy_regularization: 0.02
       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
@@ -59,8 +59,8 @@ parameters:
       memory:
         kind: 'replay'
         parameters:
-          size: 2048
-          batch_size: 256
+          size: 1024
+          batch_size: 128
           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
@@ -60,8 +60,8 @@ parameters:
       memory:
         kind: 'replay'
         parameters:
-          size: 2048
-          batch_size: 256
+          size: 1024
+          batch_size: 128
           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: 256
+      sample_count: 128
       policy_step_ratio: 0.2
       entropy_regularization: 0.02
       rollback_ratio: 0.00