TorchOptimizer

class TorchOptimizer(optimizer_class, **kwargs)

Bases: Optimizer

A wrapper class for using PyTorch optimizers.

Initialization of the TorchOptimizer class.

Parameters:

• optimizer_class (torch.optim.Optimizer) – A torch.optim.Optimizer class.

• kwargs (dict) – Additional parameters passed to optimizer_class, see more here.

hook(parameters)

Initialize the optimizer instance with the given parameters.

Parameters:

parameters (dict) – The parameters of the model to be optimized.

property instance

Get the optimizer instance.

Returns:

The optimizer instance.

Return type:

torch.optim.Optimizer