_LRSchedulers getstate include optimizer info (pytorch#7757)

* getstate should include optimizer * remove getstate/setstate functions
Tianshu-Bao · May 23, 2018 · 5a3f781 · 5a3f781
1 parent e3e15b5
commit 5a3f781
Showing 1 changed file with 0 additions and 6 deletions.
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`.