Skip to content

GitLab

Unverified Commit bfeff6cc authored by Cathy's avatar Cathy Committed by GitHub
Browse files

Fixed a typo in legacy seq2seq_trainer.py (#16531)

parent 5807054b
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/legacy/seq2seq/seq2seq_trainer.py
View file @ bfeff6cc
...@@ -115,7 +115,7 @@ class Seq2SeqTrainer(Trainer): ...@@ -115,7 +115,7 @@ class Seq2SeqTrainer(Trainer):
"eps": self.args.adam_epsilon, "eps": self.args.adam_epsilon,
} }
optimizer_kwargs["lr"] = self.args.learning_rate optimizer_kwargs["lr"] = self.args.learning_rate
if self.sharded_dpp: if self.sharded_ddp:
self.optimizer = OSS( self.optimizer = OSS(
params=optimizer_grouped_parameters, params=optimizer_grouped_parameters,
optim=optimizer_cls, optim=optimizer_cls,
......
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