From 01fee7d37a116265edb0f16e0b2f75d2116eb9f6 Mon Sep 17 00:00:00 2001 From: Volpeon Date: Wed, 4 Jan 2023 12:18:07 +0100 Subject: Various updates --- training/optimization.py | 4 ---- 1 file changed, 4 deletions(-) (limited to 'training') 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 import torch from torch.optim.lr_scheduler import LambdaLR -from diffusers.utils import logging - -logger = logging.get_logger(__name__) - class OneCyclePhase(NamedTuple): step_min: int -- cgit v1.2.3-70-g09d2