Skip to content

Commit

Permalink
Change logging level from warning to info for max_steps overriding …
Browse files Browse the repository at this point in the history
…`num_train_epochs` (#34810)

Update trainer.py
  • Loading branch information
qgallouedec authored Nov 21, 2024
1 parent 28fb02f commit d4e1acb
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/transformers/trainer.py
Original file line number Diff line number Diff line change
Expand Up @@ -661,7 +661,7 @@ def __init__(
raise ValueError("The `data_collator` should be a simple callable (function, class with `__call__`).")

if args.max_steps > 0 and args.num_train_epochs > 0:
logger.warning("max_steps is given, it will override any value given in num_train_epochs")
logger.info("max_steps is given, it will override any value given in num_train_epochs")

if train_dataset is not None and not has_length(train_dataset) and args.max_steps <= 0:
raise ValueError(
Expand Down

0 comments on commit d4e1acb

Please sign in to comment.