def main():
cfg = TrainOptions().parse() # get training options
cfg.NUM_GPUS = torch.cuda.device_count()
cfg.batch_size = int(cfg.batch_size / max(1, cfg.NUM_GPUS))
cfg.phase = 'train'
launch_job(cfg=cfg, init_method=cfg.init_method, func=train)
