-
Notifications
You must be signed in to change notification settings - Fork 1
/
config.json
62 lines (62 loc) · 1.66 KB
/
config.json
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
{
"settings": {
"DEVICE": "cpu",
"ASSESS_FREQ": 200,
"LOG_FREQ": 1000,
"SAVE_FREQ": 1000,
"PERF_MRY": 100,
"toy_corewar": {
"NUM_ACTIONS": 225,
"NUM_REGISTERS": 4,
"MAX_LENGTH": 5,
"N_INSTRUCTIONS": 4,
"N_VARS": 3,
"N_VALS": 20,
"N_TARGETS": 4
}
},
"presets": {
"DQN": {
"agent": "DQN_Agent",
"parameters": {
"h_size": 52,
"middle_size": 128,
"lstm_layers": 2,
"learning_starts": 4759,
"learning_freq": 5,
"target_update_freq": 1000,
"lr": 0.00413,
"gamma": 0.8270825,
"batch_size": 198,
"replay_buffer_size": 100000,
"epsilon_decay_steps": 0.7,
"final_epsilon": 0.1
}
},
"PG": {
"agent": "PG_Agent",
"parameters": {
"h_size": 35,
"middle_size": 84,
"lstm_layers": 2,
"learning_starts": 100,
"learning_freq": 4,
"target_update_freq": 3220,
"lr": 0.01,
"gamma": 0.99,
"batch_size": 32,
"replay_buffer_size": 100000
}
},
"AC": {
"agent": "AC_Agent",
"parameters": {
"h_size": 50,
"middle_size": 128,
"lstm_layers": 1,
"lr": 0.001,
"gamma": 0.99
}
}
}
}