Method state_dict

ML/src/python/neuralforge/optim/optimizers.py:261–266

(self)

        return loss
    
    def state_dict(self):
        return {
            'state': self.state,
            'optimizer': self.optimizer.state_dict(),
            'param_groups': self.param_groups,
        }