From 1ae2595cc3d57741565f24eaa23987f2e4f53f6d Mon Sep 17 00:00:00 2001 From: Yury Hayeu Date: Tue, 23 Apr 2024 20:55:57 +0200 Subject: [PATCH] Change training schedule --- .../jsp/GRAPH-NN/experiments/2 (DQN)/experiment.yml | 4 ++-- .../jsp/GRAPH-NN/experiments/5 (PPO)/experiment.yml | 3 +++ .../experiments/jsp/GRAPH-NN/flexible_ppo_machine.yml | 4 ++-- .../configuration/experiments/jsp/GRAPH-NN/ppo_machine.yml | 4 ++-- .../experiments/jsp/MARL-DQN/experiment/4/machine.yml | 4 ++-- 5 files changed, 11 insertions(+), 8 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 index 1a173cc..00b578e 100644 --- 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 @@ -18,8 +18,8 @@ graph: &graph is_machine_set_in_work_center_connected: False is_work_center_set_in_shop_floor_connected: False -default_mods: &default_mods [] -# - ['util/infrastructure/cuda.yml'] +default_mods: &default_mods + - ['util/infrastructure/cuda.yml'] ############################################################################################### 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 index b5f1a0e..4471dbb 100644 --- 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 @@ -36,6 +36,7 @@ ppo_1: &ppo_1 '__none__', # 'util/agent/multi_agent.yml' ] + - [ 'util/infrastructure/cuda.yml'] ppo_2: &ppo_2 base_path: 'configuration/experiments/jsp/GRAPH-NN/flexible_machine.yml' @@ -53,6 +54,7 @@ ppo_2: &ppo_2 '__none__', # 'util/agent/multi_agent.yml' ] + - [ 'util/infrastructure/cuda.yml'] ppo_3: &ppo_3 base_path: *base_model @@ -70,6 +72,7 @@ ppo_3: &ppo_3 '__none__', # 'util/agent/multi_agent.yml' ] + - [ 'util/infrastructure/cuda.yml'] ############################################################################################### 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 46e68d4..1a5b428 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 @@ -74,8 +74,8 @@ parameters: memory: kind: 'prioritized_replay' parameters: - size: 8 - batch_size: 8 + size: 4 + batch_size: 4 prefetch: 1 return: 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 fadb25f..df9cba4 100644 --- a/diploma_thesis/configuration/experiments/jsp/GRAPH-NN/ppo_machine.yml +++ b/diploma_thesis/configuration/experiments/jsp/GRAPH-NN/ppo_machine.yml @@ -74,8 +74,8 @@ parameters: memory: kind: 'prioritized_replay' parameters: - size: 8 - batch_size: 8 + size: 4 + batch_size: 4 prefetch: 1 return: diff --git a/diploma_thesis/configuration/experiments/jsp/MARL-DQN/experiment/4/machine.yml b/diploma_thesis/configuration/experiments/jsp/MARL-DQN/experiment/4/machine.yml index 9956b81..7ada696 100644 --- a/diploma_thesis/configuration/experiments/jsp/MARL-DQN/experiment/4/machine.yml +++ b/diploma_thesis/configuration/experiments/jsp/MARL-DQN/experiment/4/machine.yml @@ -78,8 +78,8 @@ parameters: memory: kind: 'prioritized_replay' parameters: - size: 8 - batch_size: 8 + size: 4 + batch_size: 4 prefetch: 1 return: