summaryrefslogtreecommitdiffstats
path: root/training
diff options
context:
space:
mode:
Diffstat (limited to 'training')
-rw-r--r--training/optimization.py4
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
5import torch 5import torch
6from torch.optim.lr_scheduler import LambdaLR 6from torch.optim.lr_scheduler import LambdaLR
7 7
8from diffusers.utils import logging
9
10logger = logging.get_logger(__name__)
11
12 8
13class OneCyclePhase(NamedTuple): 9class OneCyclePhase(NamedTuple):
14 step_min: int 10 step_min: int