{ "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 } } }