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 | ||