Skip to content
Snippets Groups Projects
Commit 0a201e6f authored by Schneider Leo's avatar Schneider Leo
Browse files

add : ray opti

parent 2217fd04
No related branches found
No related tags found
No related merge requests found
......@@ -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']),
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment