lightning_template.utils.optim.configure_optimizers

Module Contents

Functions

parser_optim_config(optim_config)	Parse the optimizer config.
get_parameters(model, all_required_parameters)	Get all optimizer parameters.
construct_lr_scheduler(lr_scheduler, optimizer)	Constructs the lr_scheduler.
get_configure_optimizers_method(optim_config)

lightning_template.utils.optim.configure_optimizers.parser_optim_config(optim_config)

Parse the optimizer config.

Parameters:

optim_config (dict) – The optimizer and lr_scheduler config.

lightning_template.utils.optim.configure_optimizers.get_parameters(model, all_required_parameters)

Get all optimizer parameters.

Parameters:

• model – a LightningModule.

• all_required_parameters – a set of required parameter names.

lightning_template.utils.optim.configure_optimizers.construct_lr_scheduler(lr_scheduler, optimizer)

Constructs the lr_scheduler.

Parameters:

• lr_scheduler – dictionary containing lr_scheduler configuration.

• optimizer – optimizer used to construct lr_scheduler.

lightning_template.utils.optim.configure_optimizers.get_configure_optimizers_method(optim_config)