File size: 998 Bytes
32211e9 | 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 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 | {
"framework": "pytorch",
"task": "text2text-generation",
"model": {
"type": "T5",
"language": "zh"
},
"pipeline": {
"type": "text2text-generation"
},
"preprocessor": {
"type": "text-gen-tokenizer"
},
"train": {
"work_dir": "/tmp",
"max_epochs": 5,
"dataloader": {
"batch_size_per_gpu": 32,
"workers_per_gpu": 0
},
"optimizer": {
"type": "AdamW",
"lr": 2e-5,
"options": {}
},
"lr_scheduler": {
"type": "LinearLR",
"start_factor": 1.0,
"end_factor": 0.0,
"total_iters": null,
"options": {
"by_epoch": false
}
},
"hooks": [
{
"type": "TextLoggerHook",
"interval": 1
},
{
"type": "IterTimerHook"
}
]
},
"evaluation": {
"dataloader": {
"batch_size_per_gpu": 32,
"workers_per_gpu": 0,
"shuffle": false
},
"metrics": [{
"type": "text-gen-metric"
}]
}
} |