SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

remove redundant code in trainer (#36994)

Browse Source 
* Update optimization.py

* Update optimization.py
This commit is contained in:
hoshi-hiyouga
2025-03-27 18:35:15 +08:00
committed by GitHub
parent d13c390d01
commit 279c2e302a
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/transformers/optimization.py
View File

@@ -582,6 +582,7 @@ def get_scheduler(
if name == SchedulerType.INVERSE_SQRT: if name == SchedulerType.INVERSE_SQRT:
return schedule_func(optimizer, num_warmup_steps=num_warmup_steps) 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: if name == SchedulerType.WARMUP_STABLE_DECAY:
return schedule_func( return schedule_func(
optimizer, optimizer,