mmengine

Commit Graph

Author	SHA1	Message	Date
Mashiro	8b0c9c5f6f	[Fix] fix build train_loop during test (#295 ) * fix build train_loop during test * fix build train_loop during test * fix build train_loop during test * fix build train_loop during test * Fix as comment	2022-06-13 21:23:46 +08:00
Mashiro	f04fec736d	[Feature]: add base model, ddp model wrapper and unit test (#268 ) * 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	2022-06-07 22:13:53 +08:00
Mashiro	3e3866c1b9	[Feature] Add optimizer wrapper (#265 ) * 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>	2022-06-01 18:04:38 +08:00
Zaida Zhou	f1da9a1d7f	[Feature] Support multiple optimizers (#235 ) * 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	2022-05-31 16:54:39 +08:00
RangiLyu	4705e1fe3d	[Enhance] Add RuntimeInfoHook to update runtime information. (#254 ) * [Enhance] Add RuntimeInfoHook to update runtime information. * move lr to runtime info * docstring * resolve comments * update ut and doc	2022-05-26 14:35:37 +08:00

5 Commits (08602a238508b70fe42e2018ecad42d24f18f938)