File size: 471 Bytes
64e04bb
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
{
  "data_path": "path/to/your/data/file",
  "num_epochs": 3,
  "cutoff_len": 512,
  "group_by_length": true,
  "output_dir": "path/to/your/output/directory",
  "batch_size": 8,
  "micro_batch_size": 1,
  "lr": 5e-5,
  "warmup_steps": 100,
  "weight_decay": 0.01,
  "adam_epsilon": 1e-8,
  "max_grad_norm": 1.0,
  "logging_steps": 50,
  "save_steps": 500,
  "eval_steps": 500,
  "overwrite_output_dir": true,
  "do_train": true,
  "do_eval": true,
  "do_predict": true
}