24bf6ad607b491fc28c83b930e7cec927599d73f,python/ray/util/sgd/torch/examples/cifar_pytorch_example.py,,,#,163

Before Change

    ray.init(address=args.address, log_to_driver=True)

    if args.tune:
        tune_example(
            num_workers=args.num_workers,
            use_gpu=args.use_gpu,
            test_mode=args.smoke_test)
    else:
        train_example(
            num_workers=args.num_workers,
            num_epochs=args.num_epochs,
            use_gpu=args.use_gpu,
            use_fp16=args.fp16,
            test_mode=args.smoke_test)

After Change

        pbar.set_postfix(dict(acc=val_stats["val_accuracy"]))

    print(trainer1.validate())
    trainer1.shutdown()
    print("success!")

Instances