fix torch warning
This commit is contained in:
@@ -114,7 +114,7 @@ class DeepSADTrainer(BaseTrainer):
|
|||||||
if epoch in self.lr_milestones:
|
if epoch in self.lr_milestones:
|
||||||
logger.info(
|
logger.info(
|
||||||
" LR scheduler: new learning rate is %g"
|
" LR scheduler: new learning rate is %g"
|
||||||
% float(scheduler.get_lr()[0])
|
% float(scheduler.get_last_lr()[0])
|
||||||
)
|
)
|
||||||
|
|
||||||
# log epoch statistics
|
# log epoch statistics
|
||||||
|
|||||||
Reference in New Issue
Block a user