Skip to content

Commit 1212bf4

Browse files
committed
Change training schedule
1 parent 07cda6e commit 1212bf4

File tree

10 files changed

+35
-35
lines changed

10 files changed

+35
-35
lines changed

diploma_thesis/configuration/experiments/jsp/GRAPH-NN/experiments/2 (DQN)/experiment.yml

+7-7
Original file line numberDiff line numberDiff line change
@@ -315,13 +315,13 @@ task:
315315
# - *marl_dqn_2
316316
- *dqn_3
317317
# - *marl_dqn_4
318-
- output_dir: 'GAT'
319-
machine_agent:
320-
parameters:
321-
- *dqn_1_gat
322-
# - *marl_dqn_2_gin
323-
- *dqn_3_gat
324-
# - *marl_dqn_4_gin
318+
# - output_dir: 'GAT'
319+
# machine_agent:
320+
# parameters:
321+
# - *dqn_1_gat
322+
## - *marl_dqn_2_gin
323+
# - *dqn_3_gat
324+
## - *marl_dqn_4_gin
325325
- output_dir: 'SAGE'
326326
machine_agent:
327327
parameters:

diploma_thesis/configuration/experiments/jsp/GRAPH-NN/experiments/2 (PPO)/experiment.yml

+4-4
Original file line numberDiff line numberDiff line change
@@ -198,10 +198,10 @@ task:
198198
machine_agent:
199199
parameters:
200200
*ppo_1
201-
- output_dir: 'GAT'
202-
machine_agent:
203-
parameters:
204-
*ppo_2
201+
# - output_dir: 'GAT'
202+
# machine_agent:
203+
# parameters:
204+
# *ppo_2
205205
- output_dir: 'GCN'
206206
machine_agent:
207207
parameters:

diploma_thesis/configuration/experiments/jsp/GRAPH-NN/experiments/5 (DQN)/experiment.yml

+14-14
Original file line numberDiff line numberDiff line change
@@ -307,20 +307,20 @@ task:
307307
values:
308308
values:
309309
__concat__:
310-
# - output_dir: 'Baseline'
311-
# machine_agent:
312-
# parameters:
313-
# - *dqn_1
314-
## - *marl_dqn_2
315-
# - *dqn_3
316-
## - *marl_dqn_4
317-
# - output_dir: 'All'
318-
# machine_agent:
319-
# parameters:
320-
# - *dqn_1_all
321-
## - *marl_dqn_2_all
322-
# - *dqn_3_all
323-
## - *marl_dqn_4_all
310+
- output_dir: 'Baseline'
311+
machine_agent:
312+
parameters:
313+
- *dqn_1
314+
# - *marl_dqn_2
315+
- *dqn_3
316+
# - *marl_dqn_4
317+
- output_dir: 'All'
318+
machine_agent:
319+
parameters:
320+
- *dqn_1_all
321+
# - *marl_dqn_2_all
322+
- *dqn_3_all
323+
# - *marl_dqn_4_all
324324
- output_dir: 'Flexible'
325325
machine_agent:
326326
parameters:

diploma_thesis/configuration/experiments/jsp/GRAPH-NN/experiments/5 (PPO)/experiment.yml

+4-4
Original file line numberDiff line numberDiff line change
@@ -206,10 +206,10 @@ task:
206206
machine_agent:
207207
parameters:
208208
*ppo_3
209-
- output_dir: 'Flexible GAT'
210-
machine_agent:
211-
parameters:
212-
*ppo_4
209+
# - output_dir: 'Flexible GAT'
210+
# machine_agent:
211+
# parameters:
212+
# *ppo_4
213213

214214
tape:
215215
machine_reward:

diploma_thesis/configuration/experiments/jsp/GRAPH-NN/flexible_machine.yml

+1-1
Original file line numberDiff line numberDiff line change
@@ -60,7 +60,7 @@ parameters:
6060
kind: 'replay'
6161
parameters:
6262
size: 2048
63-
batch_size: 192
63+
batch_size: 256
6464
prefetch: 8
6565

6666
loss:

diploma_thesis/configuration/experiments/jsp/GRAPH-NN/flexible_marl_machine.yml

+1-1
Original file line numberDiff line numberDiff line change
@@ -61,7 +61,7 @@ parameters:
6161
kind: 'replay'
6262
parameters:
6363
size: 2048
64-
batch_size: 192
64+
batch_size: 256
6565
prefetch: 8
6666

6767
loss:

diploma_thesis/configuration/experiments/jsp/GRAPH-NN/flexible_ppo_machine.yml

+1-1
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,7 @@ parameters:
4949
kind: 'ppo'
5050
parameters:
5151
device: 'cpu'
52-
sample_count: 192
52+
sample_count: 256
5353
policy_step_ratio: 0.2
5454
entropy_regularization: 0.01
5555
rollback_ratio: 0.00

diploma_thesis/configuration/experiments/jsp/GRAPH-NN/machine.yml

+1-1
Original file line numberDiff line numberDiff line change
@@ -60,7 +60,7 @@ parameters:
6060
kind: 'replay'
6161
parameters:
6262
size: 2048
63-
batch_size: 192
63+
batch_size: 256
6464
prefetch: 8
6565

6666
loss:

diploma_thesis/configuration/experiments/jsp/GRAPH-NN/marl_machine.yml

+1-1
Original file line numberDiff line numberDiff line change
@@ -61,7 +61,7 @@ parameters:
6161
kind: 'replay'
6262
parameters:
6363
size: 2048
64-
batch_size: 192
64+
batch_size: 256
6565
prefetch: 8
6666

6767
loss:

diploma_thesis/configuration/experiments/jsp/GRAPH-NN/ppo_machine.yml

+1-1
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,7 @@ parameters:
4949
kind: 'ppo'
5050
parameters:
5151
device: 'cpu'
52-
sample_count: 192
52+
sample_count: 256
5353
policy_step_ratio: 0.2
5454
entropy_regularization: 0.01
5555
rollback_ratio: 0.00

0 commit comments

Comments
 (0)