efd31f17f6753ee92ef1038016bd9470f25867d1,train.py,,,#,74

Before Change

    if not os.path.exists("./saves"):
        os.makedirs("./saves")
    // Device
    args.device = torch.device(args.device
        if torch.cuda.is_available() else "cpu")
    // Slurm
    if "SLURM_JOB_ID" in os.environ:
        args.output_folder += "-{0}".format(os.environ["SLURM_JOB_ID"])

After Change

    misc.add_argument("--num-workers", type=int, default=mp.cpu_count() - 1,
        help="number of workers for trajectories sampling (default: "
             "{0})".format(mp.cpu_count() - 1))
    misc.add_argument("--use-cuda", action="store_true",
        help="use cuda (default: false, use cpu)")

    args = parser.parse_args()
    args.device = "cuda" if args.use_cuda else "cpu"

Instances