- Jun 28, 2022
-
-
Zaida Zhou authored
-
- Jun 21, 2022
-
-
Alex Yang authored
* [Feat] Support FSDP Training * fix version comparison * change param format and move `FSDP_WRAP_POLICY` to wrapper file * add docstring and type hint,reformat code * fix type hint * fix typo, reformat code
-
- Jun 13, 2022
-
-
Mashiro authored
* add module list * add module list * fix docstring
-
- Jun 09, 2022
-
-
Mashiro authored
* fix BaseDataPreprocessor * fix BaseDataPreprocessor * change device type to torch.device * change device type to torch.device * fix cpu method of base model
-
- 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
-
- 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
-
- Feb 26, 2022
-
-
Haian Huang(深度眸) authored
* add mmddp * update * update code * update unittest * fix comment
-