Change training schedule

yura-hb · yura-hb · commit 1ae2595cc3d5 · 2024-04-23T20:55:57.000+02:00
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:

Original file line number	Diff line number	Diff line change
`@@ -36,6 +36,7 @@ ppo_1: &ppo_1`
`36`	`36`	`'__none__',`
`37`	`37`	`# 'util/agent/multi_agent.yml'`
`38`	`38`	`]`
	`39`	`+ - [ 'util/infrastructure/cuda.yml']`
`39`	`40`
`40`	`41`	`ppo_2: &ppo_2`
`41`	`42`	`base_path: 'configuration/experiments/jsp/GRAPH-NN/flexible_machine.yml'`
`@@ -53,6 +54,7 @@ ppo_2: &ppo_2`
`53`	`54`	`'__none__',`
`54`	`55`	`# 'util/agent/multi_agent.yml'`
`55`	`56`	`]`
	`57`	`+ - [ 'util/infrastructure/cuda.yml']`
`56`	`58`
`57`	`59`	`ppo_3: &ppo_3`
`58`	`60`	`base_path: *base_model`
`@@ -70,6 +72,7 @@ ppo_3: &ppo_3`
`70`	`72`	`'__none__',`
`71`	`73`	`# 'util/agent/multi_agent.yml'`
`72`	`74`	`]`
	`75`	`+ - [ 'util/infrastructure/cuda.yml']`
`73`	`76`
`74`	`77`	`###############################################################################################`
`75`	`78`