From 8ee4f7eca36100f4c7847c33025aa4fd062493b7 Mon Sep 17 00:00:00 2001 From: Kai Fricke Date: Fri, 23 Oct 2020 20:42:13 +0100 Subject: [PATCH] [tune] fix pbt ptl example (#11573) * [tune] fix pbt ptl example * wider smoke test --- python/ray/tune/examples/mnist_pytorch_lightning.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/python/ray/tune/examples/mnist_pytorch_lightning.py b/python/ray/tune/examples/mnist_pytorch_lightning.py index 86f30ed2935f9..a70db2e59e9a4 100644 --- a/python/ray/tune/examples/mnist_pytorch_lightning.py +++ b/python/ray/tune/examples/mnist_pytorch_lightning.py @@ -248,7 +248,7 @@ def tune_mnist_pbt(num_samples=10, num_epochs=10, gpus_per_trial=0): mode="min", perturbation_interval=4, hyperparam_mutations={ - "lr": lambda: tune.loguniform(1e-4, 1e-1).func(None), + "lr": tune.loguniform(1e-4, 1e-1), "batch_size": [32, 64, 128] }) @@ -285,8 +285,8 @@ def tune_mnist_pbt(num_samples=10, num_epochs=10, gpus_per_trial=0): args, _ = parser.parse_known_args() if args.smoke_test: - tune_mnist_asha(num_samples=1, num_epochs=1, gpus_per_trial=0) - tune_mnist_pbt(num_samples=1, num_epochs=1, gpus_per_trial=0) + tune_mnist_asha(num_samples=1, num_epochs=6, gpus_per_trial=0) + tune_mnist_pbt(num_samples=1, num_epochs=6, gpus_per_trial=0) else: # ASHA scheduler tune_mnist_asha(num_samples=10, num_epochs=10, gpus_per_trial=0)