Skip to content
Snippets Groups Projects
Unverified Commit 0279ac2e authored by RangiLyu's avatar RangiLyu Committed by GitHub
Browse files

[Feature] Support EMA and SWA. (#239) 

* [Feature] Support EMA and SWA.

* add ema hook

* add avg model ut

* add more unit tests

* resolve comments

* fix warmup ema

* rename

* fix comments

* add assert

* fix typehint

* add comments
parent 86ffc19c
No related branches found
No related tags found
Hide whitespace changes
Inline Side-by-side
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

Datenschutz und Impressum