construct_optimizer#

construct_optimizer(model, optimizer_cfg)#

Constructs an optimizer for a given model given a configuration.

Parameters:
  • model – a list of parameters to be trained

  • optimizer_cfg (LazyConfig) – The optimizer configuration.

Returns:

The constructed optimizer.

Return type:

torch.optim.Optimizer