def train(gpu, args):
    exp = Workbench(args.config_yaml, ddp=True)
    exp.run()
from workbench import Workbench

if __name__ == '__main__':
    exp = Workbench("config/config.yaml")
    exp.run()