saymrwulf/transformers
1
0
Fork 0
mirror of https://github.com/saymrwulf/transformers.git synced 2026-05-14 20:58:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

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

Browse source
This commit is contained in:
Cathy 2022-04-01 15:17:31 +08:00 committed by GitHub
parent 5807054bd3
commit bfeff6cc6a
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
examples/legacy/seq2seq/seq2seq_trainer.py
View file

@ -115,7 +115,7 @@ class Seq2SeqTrainer(Trainer):
"eps": self.args.adam_epsilon,
}
optimizer_kwargs["lr"] = self.args.learning_rate
if self.sharded_dpp:
if self.sharded_ddp:
self.optimizer = OSS(
params=optimizer_grouped_parameters,
optim=optimizer_cls,