-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathconfig.py
38 lines (32 loc) · 809 Bytes
/
config.py
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
#### SELF PLAY
# EPISODES = 5
# MCTS_SIMS = 50
# MEMORY_SIZE = 30000
EPISODES = 5
# EPISODES = 75
MCTS_SIMS = 10
# MCTS_SIMS = 50
MEMORY_SIZE = 2000
# TURNS_UNTIL_TAU0 = 20 # turn on which it starts playing deterministically
TURNS_UNTIL_TAU0 = 8 # turn on which it starts playing deterministically
CPUCT = 1
EPSILON = 0.2
ALPHA = 0.8
#### RETRAINING
BATCH_SIZE = 256
EPOCHS = 1
REG_CONST = 0.0001
LEARNING_RATE = 0.1
MOMENTUM = 0.9
TRAINING_LOOPS = 10
HIDDEN_CNN_LAYERS = [
{'filters': 75, 'kernel_size': (2, 2)},
{'filters': 75, 'kernel_size': (2, 2)},
{'filters': 75, 'kernel_size': (2, 2)},
{'filters': 75, 'kernel_size': (2, 2)},
{'filters': 75, 'kernel_size': (2, 2)},
{'filters': 75, 'kernel_size': (2, 2)}
]
#### EVALUATION
EVAL_EPISODES = 20
SCORING_THRESHOLD = 1.3