Batched-Impala-PyTorch/default.cfg at master · Sheepsody/Batched-Impala-PyTorch · GitHub

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
[settings]
tensorboard = logs
device = cuda
# Should be Nt=Np=2 and Na=8xcpu
agents = 2
trainers = 1
predictors = 1
training_queue = 16
training_batch_size = 4
# Up should be 32
prediction_queue = 12
prediction_batch_size = 6
checkpoint_path = checkpoint.pt
max_nb_episodes = 100000

[impala]
sequence_length = 20
rho = 1.0
cis = 1.0
discount_factor = 0.99
value_coef = 0.5
entropy_coef = 0.01

[optimizer]
function = SGD
lr = 0.0003
alpha = 0
eps = 0.01
momentum = 0.0
weight_decay = 0
centered = False

[environnement]
stacks = 1
height = 54
width = 54

[levels]
BowserCastle.Act1 = none
BowserCastle.Act2 = none
BowserCastle.Act3 = none
ChocoIsland.Act1 = none
ChocoIsland.Act2 = none
DonutPlains.Act1 = none
DonutPlains.Act2 = none
DonutPlains.Act3 = none
GhostValley.Act1 = none
GhostValley.Act2 = none
GhostValley.Act3 = none
KoopaBeach.Act1 = none
KoopaBeach.Act2 = none
MarioCircuit.Act1 = train
MarioCircuit.Act2 = train
MarioCircuit.Act3 = train
MarioCircuit.Act4 = test
RainbowRoad.Act1 = none
VanillaLake.Act1 = none
VanillaLake.Act2 = none