57 lines
1.1 KiB
JSON
57 lines
1.1 KiB
JSON
|
{
|
||
|
|
"framework": "pytorch",
|
||
|
|
"task": "fill-mask",
|
||
|
|
"model": {
|
||
|
|
"type": "structbert",
|
||
|
|
"language": "zh"
|
||
|
|
},
|
||
|
|
"pipeline": {
|
||
|
|
"type": "fill-mask"
|
||
|
|
},
|
||
|
|
"preprocessor": {
|
||
|
|
"type": "fill-mask"
|
||
|
|
},
|
||
|
|
"train": {
|
||
|
|
"work_dir": "/tmp",
|
||
|
|
"max_epochs": 5,
|
||
|
|
"dataloader": {
|
||
|
|
"batch_size_per_gpu": 32,
|
||
|
|
"workers_per_gpu": 1
|
||
|
|
},
|
||
|
|
"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": "CheckpointHook",
|
||
|
|
"interval": 1
|
||
|
|
}, {
|
||
|
|
"type": "TextLoggerHook",
|
||
|
|
"interval": 1
|
||
|
|
}, {
|
||
|
|
"type": "IterTimerHook"
|
||
|
|
}, {
|
||
|
|
"type": "EvaluationHook",
|
||
|
|
"by_epoch": false,
|
||
|
|
"interval": 100
|
||
|
|
}]
|
||
|
|
},
|
||
|
|
"evaluation": {
|
||
|
|
"dataloader": {
|
||
|
|
"batch_size_per_gpu": 32,
|
||
|
|
"workers_per_gpu": 1,
|
||
|
|
"shuffle": false
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|