codes.tune.optuna_fcts

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)

Make Optuna suggested parameters from the optuna_config.yaml file.

training_run(trial, device, config, study_name)

Run the training for a single Optuna trial and return the loss.