diff options
Diffstat (limited to 'training/functional.py')
-rw-r--r-- | training/functional.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/training/functional.py b/training/functional.py index 7104a88..bd8cbad 100644 --- a/training/functional.py +++ b/training/functional.py | |||
@@ -524,7 +524,7 @@ def train_loop( | |||
524 | 524 | ||
525 | lr = lr_scheduler.get_last_lr()[0] | 525 | lr = lr_scheduler.get_last_lr()[0] |
526 | if torch.is_tensor(lr): | 526 | if torch.is_tensor(lr): |
527 | lr = lr.item | 527 | lr = lr.item() |
528 | 528 | ||
529 | lrs.append(lr) | 529 | lrs.append(lr) |
530 | 530 | ||