diff --git a/image_ref/main_ray.py b/image_ref/main_ray.py index c66087cb3ab2c67e121054838556b0d57253129b..90f45583da74f8374e5aca0fda7bb61c0043b812 100644 --- a/image_ref/main_ray.py +++ b/image_ref/main_ray.py @@ -212,7 +212,7 @@ def test_model(best_result, args): def main(args, gpus_per_trial=1): config = { "lr": tune.loguniform(1e-4, 1e-2), - "noise": tune.loguniform(0, 500), + "noise": tune.loguniform(1e-7, 500), "positive_prop": tune.uniform(0, 100), "optimizer": tune.choice(['Adam', 'SGD']), "sampler": tune.choice(['random', 'balanced']),