File size: 499 Bytes
eaf3978
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
{'adam_epsilon': 1e-08,
 'data_dir': '',
 'early_stop_callback': False,
 'eval_batch_size': 4,
 'fp_16': False,
 'gradient_accumulation_steps': 16,
 'learning_rate': 0.0003,
 'max_grad_norm': 1.0,
 'max_seq_length': 150,
 'model_name_or_path': 'google/mt5-base',
 'n_gpu': 1,
 'num_train_epochs': 2,
 'opt_level': 'O1',
 'output_dir': 'mt5model_dh_small/',
 'seed': 42,
 'tokenizer_name_or_path': 'google/mt5-base',
 'train_batch_size': 4,
 'warmup_steps': 0,
 'weight_decay': 0.0}