-
Notifications
You must be signed in to change notification settings - Fork 5
/
64x6.yaml
34 lines (31 loc) · 1.34 KB
/
64x6.yaml
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
%YAML 1.2
---
name: '64x6-test-6' # ideally no spaces
gpu: 0 # gpu id to process on
dataset:
num_chunks: 120000 # newest nof chunks to parse
train_ratio: 0.90 # trainingset ratio
# For separated test and train data.
input_train: '/ramdisk/train/' # supports glob
input_test: '/ramdisk/test/' # supports glob
# For a one-shot run with all data in one directory.
#input: '/work/lc0/data/'
training:
batch_size: 2048 # training batch
test_steps: 500 # eval test set values after this many steps
train_avg_report_steps: 250 # training reports its average values after this many steps.
total_steps: 2500 # terminate after these steps
# checkpoint_steps: 10000 # optional frequency for checkpointing before finish
shuffle_size: 500000 # size of the shuffle buffer
lr_values: # list of learning rates
- 0.001
- 0.001
lr_boundaries: # list of boundaries
- 100
policy_loss_weight: 1.0 # weight of policy loss
value_loss_weight: 1.0 # weight of value loss
path: '/work/lc0/networks' # network storage dir
model:
filters: 64
residual_blocks: 6
...