From 745efbb8f58decc671f845a9f5ea6d0e2723486c Mon Sep 17 00:00:00 2001 From: Jan Kowalczyk Date: Thu, 4 Jul 2024 07:39:37 +0200 Subject: [PATCH] fix torch warning --- Deep-SAD-PyTorch/src/optim/DeepSAD_trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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