Update scheduler.py

get_lr is deprecated.

Author: ildoonet

warmup_scheduler/scheduler.py

@@ -28,7 +28,7 @@ def get_lr(self):
                 if not self.finished:
                     self.after_scheduler.base_lrs = [base_lr * self.multiplier for base_lr in self.base_lrs]
                     self.finished = True
-                return self.after_scheduler.get_lr()
+                return self.after_scheduler.get_last_lr()
             return [base_lr * self.multiplier for base_lr in self.base_lrs]
 
         if self.multiplier == 1.0: