def METHOD_NAME(self):
    # Train model for 5 iterations and use GPU 1
    cfg, args = setup_args()
    args.distributed_world_size = 1
    args.warmup_iterations = 5
    cfg.distributed_training.distributed_world_size = args.distributed_world_size
    cfg.bmuf.distributed_world_size = args.distributed_world_size
    cfg.bmuf.warmup_iterations = args.warmup_iterations
    iterations = 20
    results = self.bmuf_process(cfg, args, iterations)
    assert len(results) == 1