2.1.3. emicroml.modelling.optimizers

For creating wrappers to PyTorch optimizer classes.

Classes

AdamW

A wrapper to the PyTorch optimizer class torch.optim.AdamW.

BaseMLOptimizer

A wrapper to a PyTorch optimizer class.

Generic

A generic wrapper to a PyTorch optimizer class.

SGD

A wrapper to the PyTorch optimizer class torch.optim.SGD.