86cff17e7ef1b355e1ab4862ccd664ed4c856227,python/ray/util/sgd/torch/examples/tune_example.py,,tune_example,#Any#Any#,59

Before Change

def tune_example(num_workers=1, use_gpu=False):
    config = {
        "model_creator": model_creator,
        "data_creator": data_creator,
        "optimizer_creator": optimizer_creator,
        "loss_creator": nn.MSELoss,
        "num_workers": num_workers,
        "use_gpu": use_gpu,
        "config": {"batch_size": 512 // num_workers},
        "backend": "gloo"
    }

    analysis = tune.run(
        TorchTrainable,
        num_samples=12,

After Change

    analysis = tune.run(
        TorchTrainable,
        num_samples=3,
        config={"lr": tune.grid_search([1e-4, 1e-3])},
        stop={"training_iteration": 2},
        verbose=1)

Instances