"...git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "22d159ddf93990a3340a93aa49fa96ca60bd8cc3"
Unverified Commit 574e68d5 authored by Apoorv Khandelwal's avatar Apoorv Khandelwal Committed by GitHub
Browse files

Allow `Trainer.get_optimizer_cls_and_kwargs` to be overridden (#31875)

* Change `Trainer.get_optimizer_cls_and_kwargs` to `self.`

* Make `get_optimizer_cls_and_kwargs` an instance method

* Fixing typo

* Revert `get_optimizer_cls_and_kwargs` to staticmethod

* restore newline to trainer.py eof
parent 52585019
...@@ -1079,7 +1079,7 @@ class Trainer: ...@@ -1079,7 +1079,7 @@ class Trainer:
}, },
] ]
optimizer_cls, optimizer_kwargs = Trainer.get_optimizer_cls_and_kwargs(self.args, opt_model) optimizer_cls, optimizer_kwargs = self.get_optimizer_cls_and_kwargs(self.args, opt_model)
# Overwrite `params` in case it's created by `get_optimizer_cls_and_kwargs` # Overwrite `params` in case it's created by `get_optimizer_cls_and_kwargs`
# e.g. for GaLore optimizer. # e.g. for GaLore optimizer.
......
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