    import datetime
    timeout = datetime.timedelta(seconds=100000000)
    accelerator = Accelerator(
        gradient_accumulation_steps=int(config["gradient_accumulation_steps"]),
        mixed_precision="no",

        kwargs_handlers=[DistributedDataParallelKwargs(find_unused_parameters=False),
                         InitProcessGroupKwargs(timeout=timeout)],
    )
    device = accelerator.device
    exp_save_dir = os.path.join(save_dir, get_exp_name(config))
