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