1 files changed, 1 insertions, 1 deletions
diff --git a/training/optimization.py b/training/optimization.py
index 0fd7ec8..0e603fa 100644
--- a/training/optimization.py
+++ b/training/optimization.py
@@ -6,7 +6,7 @@ from diffusers.utils import logging
 logger = logging.get_logger(__name__)
-def get_one_cycle_schedule(optimizer, num_training_steps, annealing="cos", min_lr=0.05, mid_point=0.43, last_epoch=-1):
+def get_one_cycle_schedule(optimizer, num_training_steps, annealing="cos", min_lr=0.05, mid_point=0.4, last_epoch=-1):
    """
    Create a schedule with a learning rate that decreases linearly from the initial lr set in the optimizer to 0, after
    a warmup period during which it increases linearly from 0 to the initial lr set in the optimizer.

diff --git a/training/optimization.py b/training/optimization.py index 0fd7ec8..0e603fa 100644 --- a/training/optimization.py +++ b/training/optimization.py
@@ -6,7 +6,7 @@ from diffusers.utils import logging
6	logger = logging.get_logger(__name__)	6	logger = logging.get_logger(__name__)
7		7
8		8
9	def get_one_cycle_schedule(optimizer, num_training_steps, annealing="cos", min_lr=0.05, mid_point=0.43, last_epoch=-1):	9	def get_one_cycle_schedule(optimizer, num_training_steps, annealing="cos", min_lr=0.05, mid_point=0.4, last_epoch=-1):
10	"""	10	"""
11	Create a schedule with a learning rate that decreases linearly from the initial lr set in the optimizer to 0, after	11	Create a schedule with a learning rate that decreases linearly from the initial lr set in the optimizer to 0, after
12	a warmup period during which it increases linearly from 0 to the initial lr set in the optimizer.	12	a warmup period during which it increases linearly from 0 to the initial lr set in the optimizer.