src/transformers/trainer.py · 0e6ec2a4698916ba8f26272564c1776fdabc89cf · chenpangpang / transformers

Extend Transformers Trainer Class to Enable PyTorch SGD/Adagrad Optimizers for Training (#17154) · 0e6ec2a4

jianan-gu authored May 11, 2022



* add torch SGD and Adagrad optimizer bits

* refine naming
Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>
Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>

0e6ec2a4

trainer.py 149 KB

Replace trainer.py