codes.tune.optuna_fcts

Functions

create_objective(config, study_name, ...)	Create the objective function for Optuna.
get_activation_function(name)	Get the activation function module from its name.
load_yaml_config(config_path)	Load a YAML configuration file.
make_optuna_params(trial, optuna_params)	Suggest hyperparameters from optuna_params, sampling coeff_width/layers only if coeff_network == True, converting any "True"/"False" strings into bool, and mapping any activation names into nn.Modules.
training_run(trial, device, config, study_name)	Run the training for a single Optuna trial and return the loss.