From 208e48134e324e934ad964bdc61880cc923f4c0d Mon Sep 17 00:00:00 2001 From: Volpeon Date: Sat, 1 Apr 2023 22:13:55 +0200 Subject: Revert --- training/functional.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'training/functional.py') 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( lr = lr_scheduler.get_last_lr()[0] if torch.is_tensor(lr): - lr = lr.item + lr = lr.item() lrs.append(lr) -- cgit v1.2.3-70-g09d2