CUDA cleanup.
@@ -165,6 +165,7 @@ def main(arguments):
torch.cuda.manual_seed_all(seed)
except AttributeError:
log.warning("GPU access failed. You might be using a CPU-only installation of PyTorch. Falling back to CPU.")
+ args.cuda = -1
log.info("Using random seed %d", seed)
# Prepare data #
