resettable_optimizer

Description

Wrapper around pytorch optimizer to make them resettable.

Classes

Wrapper around torch optimizers to enable reset and automatic learning rate handling.