SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

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
1 changed files with 1 additions and 1 deletions
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, "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,