forked from Green-Wood/BTTR
-
Notifications
You must be signed in to change notification settings - Fork 0
/
config.yaml
36 lines (36 loc) · 766 Bytes
/
config.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
35
36
seed_everything: 7
trainer:
checkpoint_callback: true
callbacks:
- class_path: pytorch_lightning.callbacks.LearningRateMonitor
init_args:
logging_interval: epoch
- class_path: pytorch_lightning.callbacks.ModelCheckpoint
init_args:
save_top_k: 1
monitor: val_ExpRate
mode: max
filename: '{epoch}-{step}-{val_ExpRate:.4f}'
# gpus: 1
gpus: 4
accelerator: ddp
check_val_every_n_epoch: 2
max_epochs: 300
model:
d_model: 256
growth_rate: 24
num_layers: 16
nhead: 8
num_decoder_layers: 3
dim_feedforward: 1024
dropout: 0.3
beam_size: 10
max_len: 200
alpha: 1.0
learning_rate: 1.0
patience: 20
data:
zipfile_path: data.zip
test_year: 2014
batch_size: 8
num_workers: 5