mmengine

Author	SHA1	Message	Date
Mashiro	dbae83c52f	[Enhancement] Replace warnings.warn with print_log (#961 ) * Replace warning with print_log * Add comments for testing warning	2023-03-06 17:25:28 +08:00
RangiLyu	f2b0540f58	[Enhance] Raise warning for abnormal momentum (#655 )	2022-11-18 01:17:12 +08:00
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