fix torch warning

2024-07-04 07:39:37 +02:00
parent 61424bf053
commit 745efbb8f5
1 changed files with 1 additions and 1 deletions
--- 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