File size: 449 Bytes
18a2115
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
{'batch_size_per_replica': 8,
 'data_num': -1,
 'deepspeed': None,
 'epochs': 3,
 'eval_step': 500,
 'fp16': False,
 'grad_acc_steps': 4,
 'load': 'Salesforce/codet5p-220m',
 'local_rank': -1,
 'log_freq': 10,
 'lr': 5e-05,
 'lr_warmup_steps': 200,
 'max_source_len': 1024,
 'max_target_len': 50,
 'save_dir': 'saved_models/q90cce',
 'save_freq': 500,
 'train_dataset': 'dataset/funcom_q90_train.json',
 'val_dataset': 'dataset/funcom_q90_val.json'}