Skip to content

GitLab

"tests/encoder_decoder/test_modeling_encoder_decoder.py" did not exist on "a573777901e662ec2e565be312ffaeedef6effec"
Unverified Commit 8481ecef authored by Ulaş "Sophylax" Sert's avatar Ulaş "Sophylax" Sert Committed by GitHub
Browse files

Fix Type Hint of Nan/Inf Logging Filter Arg (#16227)

parent 5a6b3ccd
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/transformers/training_args.py
View file @ 8481ecef
...@@ -520,7 +520,7 @@ class TrainingArguments: ...@@ -520,7 +520,7 @@ class TrainingArguments:
) )
logging_first_step: bool = field(default=False, metadata={"help": "Log the first global_step"}) logging_first_step: bool = field(default=False, metadata={"help": "Log the first global_step"})
logging_steps: int = field(default=500, metadata={"help": "Log every X updates steps."}) logging_steps: int = field(default=500, metadata={"help": "Log every X updates steps."})
logging_nan_inf_filter: str = field(default=True, metadata={"help": "Filter nan and inf losses for logging."}) logging_nan_inf_filter: bool = field(default=True, metadata={"help": "Filter nan and inf losses for logging."})
save_strategy: IntervalStrategy = field( save_strategy: IntervalStrategy = field(
default="steps", default="steps",
metadata={"help": "The checkpoint save strategy to use."}, metadata={"help": "The checkpoint save strategy to use."},
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment