From 279c2e302ae4993986d6681c5885990c55eb5972 Mon Sep 17 00:00:00 2001 From: hoshi-hiyouga Date: Thu, 27 Mar 2025 18:35:15 +0800 Subject: [PATCH] remove redundant code in trainer (#36994) * Update optimization.py * Update optimization.py --- src/transformers/optimization.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/transformers/optimization.py b/src/transformers/optimization.py index 14e4dd9b7cd..2e6ac0161c6 100644 --- a/src/transformers/optimization.py +++ b/src/transformers/optimization.py @@ -582,6 +582,7 @@ def get_scheduler( if name == SchedulerType.INVERSE_SQRT: return schedule_func(optimizer, num_warmup_steps=num_warmup_steps) + # wsd scheduler requires either num_training_steps or num_stable_steps if name == SchedulerType.WARMUP_STABLE_DECAY: return schedule_func( optimizer,