-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathconfig.py
45 lines (41 loc) · 1.25 KB
/
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
39
40
41
42
43
44
45
#### SELF PLAY
EPISODES = 5
MCTS_SIMS = 5
MEMORY_SIZE = 200
TURNS_UNTIL_TAU0 = 10 # 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': (4,4)}
, {'filters':75, 'kernel_size': (4,4)}
, {'filters':75, 'kernel_size': (4,4)}
, {'filters':75, 'kernel_size': (4,4)}
, {'filters':75, 'kernel_size': (4,4)}
, {'filters':75, 'kernel_size': (4,4)}
, {'filters':75, 'kernel_size': (4,4)}
, {'filters':75, 'kernel_size': (4,4)}
, {'filters':75, 'kernel_size': (4,4)}
, {'filters':75, 'kernel_size': (4,4)}
, {'filters':75, 'kernel_size': (4,4)}
, {'filters':75, 'kernel_size': (4,4)}
, {'filters':75, 'kernel_size': (4,4)}
, {'filters':75, 'kernel_size': (4,4)}
, {'filters':75, 'kernel_size': (4,4)}
, {'filters':75, 'kernel_size': (4,4)}
, {'filters':75, 'kernel_size': (4,4)}
, {'filters':75, 'kernel_size': (4,4)}
, {'filters':75, 'kernel_size': (4,4)}
, {'filters':75, 'kernel_size': (4,4)}
, {'filters':75, 'kernel_size': (4,4)}
]
#### EVALUATION
EVAL_EPISODES = 20
SCORING_THRESHOLD = 1.3