Optimizers

class easypl.optimizers.wrapper.WrapperOptimizer(optimizer_cls, **kwargs)

Wrapper for pytorch Optimizer class.

optimizer_cls

Pytorch Optimizer class.

kwargs

Additional arguments.

__call__(params) → Optimizer

Return optimizer.

params

Model params

Returns:

Optimizer object

Return type:

Optimizer