diff --git a/Deep-SAD-PyTorch/src/optim/DeepSAD_trainer.py b/Deep-SAD-PyTorch/src/optim/DeepSAD_trainer.py
index 57d273d..b9bd7cf 100644
--- a/Deep-SAD-PyTorch/src/optim/DeepSAD_trainer.py
+++ b/Deep-SAD-PyTorch/src/optim/DeepSAD_trainer.py
@@ -114,7 +114,7 @@ class DeepSADTrainer(BaseTrainer):
             if epoch in self.lr_milestones:
                 logger.info(
                     "  LR scheduler: new learning rate is %g"
-                    % float(scheduler.get_lr()[0])
+                    % float(scheduler.get_last_lr()[0])
                 )
 
             # log epoch statistics