File size: 638 Bytes
ff29165
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
{
    "output_dir": "./results",
    "evaluation_strategy": "no",
    "do_eval": false,
    "eval_steps": 0,
    "log_level": "error",
    "logging_first_step": true,
    "logging_steps": 5,
    "logging_dir": "./logs",
    "save_steps": 200,
    "save_total_limit": 2,
    "num_train_epochs": 5,
    "per_device_train_batch_size": 12,
    "optim": "adamw_torch",
    "gradient_accumulation_steps": 4,
    "dataloader_drop_last": true,
    "warmup_steps": 100,
    "weight_decay": 0.01,
    "learning_rate": 5e-05,
    "fp16": true,
    "total_gpus": 1,
    "v_cpus": 6,
    "total_memory_in_gb": 23040.7890625,
    "dataset_limit": 100
}