diff options
| author | Volpeon <git@volpeon.ink> | 2023-04-01 22:13:55 +0200 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2023-04-01 22:13:55 +0200 |
| commit | 208e48134e324e934ad964bdc61880cc923f4c0d (patch) | |
| tree | c215f6c201c04b0b2d18ba0df230fb4c5e622985 /training/functional.py | |
| parent | Fix (diff) | |
| download | textual-inversion-diff-208e48134e324e934ad964bdc61880cc923f4c0d.tar.gz textual-inversion-diff-208e48134e324e934ad964bdc61880cc923f4c0d.tar.bz2 textual-inversion-diff-208e48134e324e934ad964bdc61880cc923f4c0d.zip | |
Revert
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 | ||