Function cleanup

src/script/our_train_A.py:44–45

()

    torch.cuda.set_device(rank)

def cleanup():
    dist.destroy_process_group()
    
def train(rank, world_size, conf):
    os.makedirs(conf.exp.log_dir, exist_ok=True)