43 lines
712 B
YAML
43 lines
712 B
YAML
# task
|
|
binary_data_dir: ''
|
|
work_dir: '' # experiment directory.
|
|
infer: false # infer
|
|
seed: 1234
|
|
debug: false
|
|
save_codes:
|
|
- configs
|
|
- modules
|
|
- tasks
|
|
- utils
|
|
- usr
|
|
|
|
#############
|
|
# dataset
|
|
#############
|
|
ds_workers: 1
|
|
test_num: 100
|
|
valid_num: 100
|
|
endless_ds: false
|
|
sort_by_len: true
|
|
|
|
#########
|
|
# train and eval
|
|
#########
|
|
load_ckpt: ''
|
|
save_ckpt: true
|
|
save_best: false
|
|
num_ckpt_keep: 3
|
|
clip_grad_norm: 0
|
|
accumulate_grad_batches: 1
|
|
log_interval: 100
|
|
num_sanity_val_steps: 5 # steps of validation at the beginning
|
|
check_val_every_n_epoch: 10
|
|
val_check_interval: 2000
|
|
max_epochs: 1000
|
|
max_updates: 160000
|
|
max_tokens: 31250
|
|
max_sentences: 100000
|
|
max_eval_tokens: -1
|
|
max_eval_sentences: -1
|
|
test_input_dir: ''
|