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
......@@ -520,7 +520,7 @@ class TrainingArguments:
)
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_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(
default="steps",
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