wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-07-06 05:16:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

[tune] fix pbt ptl example (#11573)

Browse Source 
* [tune] fix pbt ptl example

* wider smoke test
This commit is contained in:
Kai Fricke
2020-10-23 20:42:13 +01:00
committed by GitHub
parent 7a0184e081
commit 8ee4f7eca3
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/tune/examples/mnist_pytorch_lightning.py
+3 -3
View File
@@ -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 @@ if __name__ == "__main__":
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)