From 051b5e53c5abd59a874c5828b089d1a70b6ba06a Mon Sep 17 00:00:00 2001 From: Yury Hayeu Date: Thu, 25 Apr 2024 10:03:38 +0200 Subject: [PATCH] Fix training script --- .../experiments/jsp/GRAPH-NN/flexible_machine.yml | 4 ++-- .../experiments/jsp/GRAPH-NN/flexible_marl_machine.yml | 4 ++-- .../experiments/jsp/GRAPH-NN/flexible_ppo_machine.yml | 2 +- .../configuration/experiments/jsp/GRAPH-NN/machine.yml | 4 ++-- .../configuration/experiments/jsp/GRAPH-NN/marl_machine.yml | 4 ++-- .../configuration/experiments/jsp/GRAPH-NN/ppo_machine.yml | 2 +- 6 files changed, 10 insertions(+), 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 index e648a37c..7d1c79ac 100644 --- 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 index 9a5ea01b..3b657ba0 100644 --- 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 index bf917a65..bcd65e8e 100644 --- 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 index 3bdf1394..9ed15a26 100644 --- 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 index b76c26f1..912a1201 100644 --- 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 index 166af687..f9f15925 100644 --- 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