Skip to content

Commit 6a93635

Browse files
update deprecation msg for warmup_ratio (#42813)
* update msg * Update src/transformers/training_args.py Co-authored-by: Benjamin Bossan <BenjaminBossan@users.noreply.github.com> --------- Co-authored-by: Benjamin Bossan <BenjaminBossan@users.noreply.github.com>
1 parent 0c18820 commit 6a93635

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/transformers/training_args.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -860,7 +860,7 @@ class TrainingArguments:
860860
warmup_ratio: float | None = field(
861861
default=None,
862862
metadata={
863-
"help": "This argument is deprecated and will be removed in v5. Use `warmup_steps` instead as it also works with float values."
863+
"help": "This argument is deprecated and will be removed in v5.2. Use `warmup_steps` instead as it also works with float values."
864864
},
865865
)
866866

@@ -2575,7 +2575,7 @@ def set_lr_scheduler(
25752575
```
25762576
"""
25772577
if warmup_ratio is not None:
2578-
logger.warning("warmup_ratio is deprecated and will be removed in v5. Use `warmup_steps` instead.")
2578+
logger.warning("warmup_ratio is deprecated and will be removed in v5.2 . Use `warmup_steps` instead.")
25792579
warmup_steps = warmup_ratio
25802580

25812581
self.lr_scheduler_type = SchedulerType(name)

0 commit comments

Comments
 (0)