- Sep 09, 2022
- Aug 26, 2022
-
-
RangiLyu authored
* [Fix] Fix EMAHook trigger train loop init during testing. * fix sync buffer * update ut * fix sync buffer * fix sync buffer
-
- Aug 23, 2022
-
-
Zaida Zhou authored
* Move the design of hook to design/hook.md * add relative links in docs * update docstring of hooks * refine checkpointhook docs * refine * fix comments * refine * add logging.md link in hook.md * resolve comments * fix typo
-
- Aug 08, 2022
-
-
Mashiro authored
* BaseAveragedModel support load ckpt without module prefix * refine docstring * allow EMAHook does not load ckpt strictly * add unit test for strict argument of EMAHook * sync remote * sync remote * clean the code * ema hook supports setting start iter * fix unit test * fix as comment * fix as comment * describe kwargs
-
- Jul 04, 2022
-
-
Mashiro authored
* Fix ema hook and add unit test * save state_dict of ema.module save state_dict of ema.module * replace warning.warn with MMLogger.warn * fix as comment * fix bug * fix bug
-
- May 30, 2022
-
-
RangiLyu authored
* [Fix] Fix ema state dict swapping in EMAHook. * fix pt1.5 ut * add more comments
-
- May 26, 2022
-
-
RangiLyu authored
* [Enhance] Add RuntimeInfoHook to update runtime information. * move lr to runtime info * docstring * resolve comments * update ut and doc
-
- May 19, 2022
-
-
RangiLyu authored
* [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
-