diff --git a/paddlehub/finetune/trainer.py b/paddlehub/finetune/trainer.py index 02d657e354e45f5b23b30ec3cc6f3b281388969f..860623a241cfcab41061d95ea3c314cf7bfc5b47 100644 --- a/paddlehub/finetune/trainer.py +++ b/paddlehub/finetune/trainer.py @@ -347,7 +347,7 @@ class Trainer(object): loss(paddle.Tensor) : Loss tensor. ''' self.optimizer.step() - self.learning_rate_step(epoch_idx, batch_idx, self.optimizer.get_lr(), loss) + self.learning_rate_step(epoch_idx, batch_idx, self.optimizer._learning_rate, loss) def learning_rate_step(self, epoch_idx: int, batch_idx: int, learning_rate: Generic, loss: paddle.Tensor): if isinstance(learning_rate, paddle.optimizer.lr.LRScheduler):