code: def get_state(self):
          "Return the inner state minus the layer groups."
          return {'opt_state':self.opt.state_dict(), 'lr':self._lr, 'wd':self._wd, 'beta':self._beta, 'mom':self._mom,
                  'opt_func':self.opt_func, 'true_wd':self.true_wd, 'bn_wd':self.bn_wdf get_state(self):
          "Return the inner state minus the layer groups."
          return {'opt_state':self.opt.state_dict(), 'lr':self._lr, 'wd':self._wd, 'beta':self._beta, 'mom':self._mom,
                  'opt_func':self.opt_func, 'true_wd':self.true_wd, 'bn_wd':self.bn_wd}
docs: Return the inner state minus the layer groups.