3 Commits (a6dfde5e13016c6fff3e04809afc5146cf33f571)

Author	SHA1	Message	Date
Mashiro	312f264ecd	[Feature] Add autocast wrapper (#307) ... * add autocast wrapper * fix docstring * fix docstring * fix compare version * fix unit test * fix incompatible arguments * fix as comment * fix unit test * rename auto_cast to autocast	2022-06-22 19:49:20 +08:00
RangiLyu	172b9ded4a	[Fix] Fix ema state dict swapping in EMAHook and torch1.5 ut. (#266) ... * [Fix] Fix ema state dict swapping in EMAHook. * fix pt1.5 ut * add more comments	2022-05-30 16:51:06 +08:00
RangiLyu	0279ac2e8d	[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	2022-05-19 18:53:04 +08:00