Commit 12428f0e authored by Sylvain Gugger's avatar Sylvain Gugger
Browse files

Fix style

parent 1dfc11e9
......@@ -416,7 +416,11 @@ class DefaultFlowCallback(TrainerCallback):
control.should_log = True
# Evaluate
if args.evaluation_strategy == IntervalStrategy.STEPS and state.global_step % args.eval_steps == 0 and args.eval_delay > state.global_step:
if (
args.evaluation_strategy == IntervalStrategy.STEPS
and state.global_step % args.eval_steps == 0
and args.eval_delay > state.global_step
):
control.should_evaluate = True
# Save
......
......@@ -139,7 +139,8 @@ class TrainingArguments:
left unset, the whole predictions are accumulated on GPU/TPU before being moved to the CPU (faster but
requires more memory).
eval_delay (`float`, *optional*):
Number of epochs or steps to wait for before the first evaluation can be performed, depending on the evaluation_strategy.
Number of epochs or steps to wait for before the first evaluation can be performed, depending on the
evaluation_strategy.
learning_rate (`float`, *optional*, defaults to 5e-5):
The initial learning rate for [`AdamW`] optimizer.
weight_decay (`float`, *optional*, defaults to 0):
......@@ -476,7 +477,9 @@ class TrainingArguments:
eval_delay: Optional[float] = field(
default=0,
metadata={"help": "Number of epochs or steps to wait for before the first evaluation can be performed, depending on the evaluation_strategy."},
metadata={
"help": "Number of epochs or steps to wait for before the first evaluation can be performed, depending on the evaluation_strategy."
},
)
learning_rate: float = field(default=5e-5, metadata={"help": "The initial learning rate for AdamW."})
......
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