Change training schedule

yura-hb · Apr 23, 2024 · 1ae2595 · 1ae2595
1 parent c1542b3
commit 1ae2595
Show file tree

Hide file tree

Showing 5 changed files with 11 additions and 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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -78,8 +78,8 @@ parameters:
       memory:
         kind: 'prioritized_replay'
         parameters:
-          size: 8
-          batch_size: 8
+          size: 4
+          batch_size: 4
           prefetch: 1
 
       return: