8b3f79f3079834f00c7ec62d5decd6bc026c1bc8,python/ray/tune/examples/tune_cifar10_gluon.py,,,#,186

Before Change

    import ray
    from ray.tune.schedulers import AsyncHyperBandScheduler, FIFOScheduler

    ray.init()
    if args.scheduler == "fifo":
        sched = FIFOScheduler()
    elif args.scheduler == "asynchyperband":
        sched = AsyncHyperBandScheduler(

After Change

    args = parser.parse_args()
    sched = create_scheduler(args.scheduler)

    analysis = tune.run(
        train_cifar10,
        name=args.expname,
        verbose=2,
        scheduler=sched,
        stop={
            "mean_accuracy": 0.98,
            "training_iteration": 1 if args.smoke_test else args.epochs
        },
        resources_per_trial={
            "cpu": int(args.num_workers),
            "gpu": int(args.num_gpus)
        },
        num_samples=1 if args.smoke_test else args.num_samples,
        config={
            "args": args,
            "lr": tune.loguniform(1e-4, 1e-1),
            "momentum": tune.uniform(0.85, 0.95),
        })
    print("Best hyperparameters found were: ", analysis.best_config)

Instances