5bc2ba38fd6dc54bb54c5af5b06fd40f37c7315a,python/ray/util/sgd/torch/examples/train_example.py,,,#,98

Before Change

    import ray

    ray.init(address=args.address)
    train_example(num_workers=args.num_workers, use_gpu=args.use_gpu)

After Change

    args, _ = parser.parse_known_args()

    import ray
    if args.smoke_test:
        ray.init(num_cpus=2)
    else:
        ray.init(address=args.address)
    train_example(num_workers=args.num_workers, use_gpu=args.use_gpu)

Instances