_LRSchedulers getstate include optimizer info (pytorch#7757)

ailzhang · weiyangfb · commit fc8882cb1554 · 2018-06-11T12:17:18.000-07:00
* getstate should include optimizer

* remove getstate/setstate functions
diff --git a/torch/optim/lr_scheduler.py b/torch/optim/lr_scheduler.py
@@ -22,12 +22,6 @@ def __init__(self, optimizer, last_epoch=-1):
         self.step(last_epoch + 1)
         self.last_epoch = last_epoch
 
-    def __getstate__(self):
-        return self.state_dict()
-
-    def __setstate__(self, state):
-        self.load_state_dict(state)
-
     def state_dict(self):
         """Returns the state of the scheduler as a :class:`dict`.
 
