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
Zaida Zhou
86ffc19c9c
Add pyupgrade pre-commit hook ( #232 )
...
* Add pyupgrade pre-commit hook
* fix ut
* remove comments
2022-05-19 17:56:31 +08:00
RangiLyu
3d830a28b6
[Fix]: Fix is_model_wrapper and add DistSamplerSeedHook to default hooks. ( #172 )
...
* [Fix]: Fix model_wrapper and add DistSamplerSeedHook as default hook.
* add comments
2022-04-08 22:18:23 +08:00
Haian Huang(深度眸)
5170676a2f
fix mmdp unittest ( #60 )
2022-02-27 20:34:52 +08:00
Haian Huang(深度眸)
492b2f2fa8
[Feature] Add MMDataParallel and MMDistributedDataParallel ( #44 )
...
* add mmddp
* update
* update code
* update unittest
* fix comment
2022-02-26 10:21:20 +08:00
