-
Notifications
You must be signed in to change notification settings - Fork 3.6k
Description
Thanks for your powerful project and it's really helpful. I 'd like to try it for my current research if everything goes well.
My problem is, how to set hyperparameters search range and run the search? I 've read the chapters 'CPU hyperparameter search' and 'Running grid search on a cluster' in your document, however, I guess it is not very clear as there is only a few lines of code in 'CPU hyperparameter search' chapter without explanation (and main_local appears in the code without declaration).
Here is my trial to change LightningTemplateModel and single_cpu_template.py to be able to perform a hyperparameter search:
- set
tunable=Truefor some params indef add_model_specific_args(parent_parser, root_dir)inLightningTemplateModel, e.g.,parser.opt_list('--learning_rate', default=0.001*8, type=float, options=[0.0001, 0.0005, 0.001, 0.005], tunable=True) - annotate
main(hyperparams)and addhyperparams.optimize_parallel_cpu( main, nb_trials=20, nb_workers=1 )
However, it doesn't seem to work. So how can I set hyperparameters search range and run the search?
Sorry if my presentation is unclear (as I 'm not a native speaker). Thanks.