- Jun 09, 2022
-
-
Yixiao Fang authored
-
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 08, 2022
- 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
-
RangiLyu authored
-
Mashiro authored
* rename datalist and support load ceph * rename datalist and support load ceph * remove check disk file path in _load_metainfo * fix rename error * fix rename error * unit test error * fix rename error * remove unnecessary code * fix lint
-
jbwang1997 authored
-
- Jun 06, 2022
-
-
jbwang1997 authored
* Add auto scale lr fucntion * Update * Update * Update * Update * Update * Update * Update * Update * Update * Update Co-authored-by:
wangjiabao1.vendor <wangjiabao@pjlab.org.cn>
-
Jiazhen Wang authored
* support custom runner * change build_runner_from_cfg * refine docstring * refine docstring
-
Haian Huang(深度眸) authored
* fix visbackend warning * fix
-
RangiLyu authored
* Modify val_interval and val_begin to be the attributes of TrainLoop. * update doc * fix lint * type hint
-
Alex Yang authored
-
- Jun 05, 2022
-
-
Mashiro authored
* fix build optimizer wrapper without type * refine logic * fix as comment * fix optim_wrapper config error in docstring and unit test * refine docstring of build_optim_wrapper
-
- Jun 01, 2022
-
-
Mashiro authored
* Support multiple optimizers * minor refinement * improve unit tests * minor fix * Update unit tests for resuming or saving ckpt for multiple optimizers * refine docstring * refine docstring * fix typo * update docstring * refactor the logic to build multiple optimizers * resolve comments * ParamSchedulers spports multiple optimizers * add optimizer_wrapper * fix comment and docstirng * fix unit test * add unit test * refine docstring * RuntimeInfoHook supports printing multi learning rates * resolve comments * add optimizer_wrapper * fix mypy * fix lint * fix OptimizerWrapperDict docstring and add unit test * rename OptimizerWrapper to OptimWrapper, OptimWrapperDict inherit OptimWrapper, and fix as comment * Fix AmpOptimizerWrapper * rename build_optmizer_wrapper to build_optim_wrapper * refine optimizer wrapper * fix AmpOptimWrapper.step, docstring * resolve confict * rename DefaultOptimConstructor * fix as comment * rename clig grad auguments * refactor optim_wrapper config * fix docstring of DefaultOptimWrapperConstructor fix docstring of DefaultOptimWrapperConstructor * add get_lr method to OptimWrapper and OptimWrapperDict * skip some amp unit test * fix unit test * fix get_lr, get_momentum docstring * refactor get_lr, get_momentum, fix as comment * fix error message Co-authored-by:
zhouzaida <zhouzaida@163.com>
-
- May 31, 2022
-
-
RangiLyu authored
-
Mashiro authored
-
Alex Yang authored
* [feat]:support display paramwise result in constructing optimizer * [fix]:fix format issue * delete unnecessary rank logic and fix format
-
Zaida Zhou authored
* Support multiple optimizers * minor refinement * improve unit tests * minor fix * Update unit tests for resuming or saving ckpt for multiple optimizers * refine docstring * refine docstring * fix typo * update docstring * refactor the logic to build multiple optimizers * resolve comments * ParamSchedulers spports multiple optimizers * refine docstring * RuntimeInfoHook supports printing multi learning rates * resolve comments * fix typo
-
Jiazhen Wang authored
* improve exception in call_hook * refine unit test * add test_call_hook * refine * update docstring and ut
-
jbwang1997 authored
* Update * Add unittest
-
- May 30, 2022
-
-
RangiLyu authored
* [Fix] Fix ema state dict swapping in EMAHook. * fix pt1.5 ut * add more comments
-
- May 27, 2022
-
-
Jingwei Zhang authored
* add the epoch/iter that begins validating * fix lint * add property and fix unit test * minor changes * fix typos and add unit test * add unit test about begin * fix docstring
-
Haian Huang(深度眸) authored
* fix 'Runner' object has no attribute 'log_buffer' * update * add train
-
- 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 25, 2022
-
-
Jiazhen Wang authored
* refine sync random seed * cancel seed param in batch-sampler
-
Haian Huang(深度眸) authored
* Add profiling tools * fix docstr * fix docstr * update * fix bug * update * update * fix error * fix mypy * uodate * merge main * fix UT
-
- May 24, 2022
-
-
Haian Huang(深度眸) authored
* move get_max_cuda_memory and set_multi_processing to a public function * fix lint * fix lint * fix lint * delete _set_multi_processing * fix error * rename
-
Jiazhen Wang authored
* support custom log processor * supplementary docs * format code
-
- May 21, 2022
-
-
RangiLyu authored
-
- May 20, 2022
-
-
RangiLyu authored
* [Fix] Fix some bugs in hooks and runner. * fix markdown * fix latex formula * resolve comments
-
- 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
-
Zaida Zhou authored
* Add pyupgrade pre-commit hook * fix ut * remove comments
-
Zaida Zhou authored
* [Fix] Replace torch distributed with mmengine dist module * minor refinement * move all_reduce_params to dist.py * add unit tests * update unit tests * fix test_logger.py * add examples
-
- May 18, 2022
-
-
RangiLyu authored
* [Enhance] Make loop related attributes to be runner's properties. * move iter and epoch to loop * resolve comments
-
- May 17, 2022
-
-
Mashiro authored
* implement parse_data_info * add unit test * fix join prefix of ann_file * fix docstring
-
Zaida Zhou authored
* Replace markdownlint with mdformat to avoid installing ruby * fix circleci
-
Mashiro authored
* fix runner dump cfg * convert dict cfg to Config
-
Zaida Zhou authored
* pull flake8 from github * pull flake8 from github
-
- May 13, 2022
-
-
liukuikun authored
-