61781b939bfaad37bc4ff506cd52183748e0de0d,pycorrector/seq2seq/train.py,,train,#Any#Any#Any#Any#Any#Any#Any#Any#Any#Any#Any#Any#Any#Any#Any#Any#Any#Any#Any#Any#Any#Any#Any#Any#Any#Any#Any#Any#Any#Any#,22

Before Change

        os.environ["CUDA_VISIBLE_DEVICES"] = str(config.gpu_id)
        if torch.cuda.is_available():
            device = torch.device("cuda:{}".format(gpu_id))
            torch.cuda.set_device(gpu_id)
        else:
            device = torch.device("cpu")
    else:
        device = torch.device("cpu")

After Change

    if gpu_id > -1:
        os.environ["CUDA_VISIBLE_DEVICES"] = str(config.gpu_id)
        if torch.cuda.is_available():
            device = torch.device("cuda")
        else:
            device = torch.device("cpu")
    else:
        device = torch.device("cpu")

Instances