diff options
| author | Volpeon <git@volpeon.ink> | 2023-01-04 12:18:07 +0100 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2023-01-04 12:18:07 +0100 |
| commit | 01fee7d37a116265edb0f16e0b2f75d2116eb9f6 (patch) | |
| tree | 6389f385191247fb3639900da0d29a3064259cb7 /training | |
| parent | Better eval generator (diff) | |
| download | textual-inversion-diff-01fee7d37a116265edb0f16e0b2f75d2116eb9f6.tar.gz textual-inversion-diff-01fee7d37a116265edb0f16e0b2f75d2116eb9f6.tar.bz2 textual-inversion-diff-01fee7d37a116265edb0f16e0b2f75d2116eb9f6.zip | |
Various updates
Diffstat (limited to 'training')
| -rw-r--r-- | training/optimization.py | 4 |
1 files changed, 0 insertions, 4 deletions
diff --git a/training/optimization.py b/training/optimization.py index 14c2bd5..dd84f9c 100644 --- a/training/optimization.py +++ b/training/optimization.py | |||
| @@ -5,10 +5,6 @@ from functools import partial | |||
| 5 | import torch | 5 | import torch |
| 6 | from torch.optim.lr_scheduler import LambdaLR | 6 | from torch.optim.lr_scheduler import LambdaLR |
| 7 | 7 | ||
| 8 | from diffusers.utils import logging | ||
| 9 | |||
| 10 | logger = logging.get_logger(__name__) | ||
| 11 | |||
| 12 | 8 | ||
| 13 | class OneCyclePhase(NamedTuple): | 9 | class OneCyclePhase(NamedTuple): |
| 14 | step_min: int | 10 | step_min: int |