| { | |
| "framework": "pytorch", | |
| "task": "visual-question-answering", | |
| "preprocessor": { | |
| "type": "mplug-tasks-preprocessor" | |
| }, | |
| "model": { | |
| "type": "mplug" | |
| }, | |
| "pipeline": { | |
| "type": "visual-question-answering" | |
| }, | |
| "train": { | |
| "work_dir": "/tmp", | |
| "max_epochs": 3, | |
| "dataloader": { | |
| "batch_size_per_gpu": 2, | |
| "workers_per_gpu": 1 | |
| }, | |
| "optimizer": { | |
| "type": "SGD", | |
| "lr": 0.01, | |
| "options": { | |
| "grad_clip": { | |
| "max_norm": 2.0 | |
| } | |
| } | |
| }, | |
| "lr_scheduler": { | |
| "type": "StepLR", | |
| "step_size": 2, | |
| "options": { | |
| "warmup": { | |
| "type": "LinearWarmup", | |
| "warmup_iters": 2 | |
| } | |
| } | |
| }, | |
| "hooks": [{ | |
| "type": "CheckpointHook", | |
| "interval": 1 | |
| }, { | |
| "type": "TextLoggerHook", | |
| "interval": 1 | |
| }, { | |
| "type": "IterTimerHook" | |
| }, { | |
| "type": "EvaluationHook", | |
| "interval": 1 | |
| }] | |
| }, | |
| "evaluation": { | |
| "dataloader": { | |
| "batch_size_per_gpu": 2, | |
| "workers_per_gpu": 1, | |
| "shuffle": false | |
| } | |
| } | |
| } | |