lightning_template.utils.optim.configure_optimizers

lightning_template.utils.optim.configure_optimizers#

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)

Module Contents#

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: set, finetune_rest=False, return_rest=False)#

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)#
Contents