Skip to content

Commit

Permalink
small fix
Browse files Browse the repository at this point in the history
  • Loading branch information
stas00 committed Jan 12, 2022
1 parent b839e80 commit e73249c
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/transformers/training_args.py
Original file line number Diff line number Diff line change
Expand Up @@ -655,7 +655,7 @@ class TrainingArguments:
default=0.0, metadata={"help": "The label smoothing epsilon to apply (zero means no label smoothing)."}
)
optim: OptimizerNames = field(
default=OptimizerNames.ADAMW_HF.value,
default=OptimizerNames.ADAMW_HF,
metadata={"help": "The optimizer to use.", "choices": [e.value for e in OptimizerNames]},
)
adafactor: bool = field(default=False, metadata={"help": "Whether or not to replace AdamW by Adafactor."})
Expand Down

0 comments on commit e73249c

Please sign in to comment.