- Jun 13, 2022
-
-
Mashiro authored
* merge context * update unit test * add docstring * fix bug in AmpOptimWrapper * add docstring for backward * add warning and docstring for accumuate gradient * fix docstring * fix docstring * add params_group method * fix as comment * fix as comment * make default_value of loss_scale to dynamic * Fix docstring * decouple should update and should no sync * rename attribute in OptimWrapper * fix docstring * fix comment * fix comment * fix as comment * fix as comment and add unit test
-
- Jun 07, 2022
-
-
Mashiro authored
* add base model, ddp model and unit test * add unit test * fix unit test * fix docstring * fix cpu unit test * refine base data preprocessor * refine base data preprocessor * refine interface of ddp module * remove optimizer hook * add forward * fix as comment * fix unit test * fix as comment * fix build optimizer wrapper * rebase main and fix unit test * stack_batch support stacking ndim tensor, add docstring for merge dict * fix lint * fix test loop * make precision_context effective to data_preprocessor * fix as comment * fix as comment * refine docstring * change collate_data output typehints * rename to_rgb to bgr_to_rgb and rgb_to_bgr * support build basemodel with built DataPreprocessor * fix as comment * fix docstring
-
- Apr 08, 2022
-
-
RangiLyu authored
* [Fix]: Fix model_wrapper and add DistSamplerSeedHook as default hook. * add comments
-
- Mar 05, 2022
-
-
Zaida Zhou authored
-
- Feb 26, 2022
-
-
Haian Huang(深度眸) authored
* add mmddp * update * update code * update unittest * fix comment
-