70c4ea191f
[Refactor]: Modify val_interval and val_begin to be the attributes of TrainLoop. ( #274 )
...
* Modify val_interval and val_begin to be the attributes of TrainLoop.
* update doc
* fix lint
* type hint
2022-06-06 11:13:25 +08:00
13606040ac
[Feat]:Add base module ( #277 )
2022-06-06 10:51:23 +08:00
80a46c4848
[Fix] fix build optimizer wrapper without type ( #272 )
...
* 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
2022-06-05 22:35:16 +08:00
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
987e5b83f9
fix wrong import ( #271 )
2022-05-31 22:59:08 +08:00
73da44805f
[Enhancement] add TypeVar type hint for ManagerMixin ( #269 )
2022-05-31 17:00:45 +08:00
b01b3ff97c
[feat]:support display paramwise result in constructing optimizer ( #262 )
...
* [feat]:support display paramwise result in constructing optimizer
* [fix]:fix format issue
* delete unnecessary rank logic and fix format
2022-05-31 16:59:46 +08:00
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
f2190de787
[Enhance] Improve Exception in call_hook ( #247 )
...
* improve exception in call_hook
* refine unit test
* add test_call_hook
* refine
* update docstring and ut
2022-05-31 11:34:30 +08:00
38b22d9e68
[Enhance] Enhance error report when a module has been registered in registery. ( #264 )
...
* Update
* Add unittest
2022-05-31 11:31:04 +08:00
172b9ded4a
[Fix] Fix ema state dict swapping in EMAHook and torch1.5 ut. ( #266 )
...
* [Fix] Fix ema state dict swapping in EMAHook.
* fix pt1.5 ut
* add more comments
2022-05-30 16:51:06 +08:00
40daf46a45
Support validation only after some epoch/iteration in ValLoop ( #257 )
...
* 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
2022-05-27 15:10:12 +08:00
08a3adb5d7
Fix error of 'Runner' object has no attribute 'log_buffer' ( #259 )
...
* fix 'Runner' object has no attribute 'log_buffer'
* update
* add train
2022-05-27 10:51:25 +08:00
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
4cbbbc0c31
[Enhance] Refine sync random seed ( #256 )
...
* refine sync random seed
* cancel seed param in batch-sampler
2022-05-25 19:18:03 +08:00
c197bdf359
[Feature] Profiling tools ( #241 )
...
* Add profiling tools
* fix docstr
* fix docstr
* update
* fix bug
* update
* update
* fix error
* fix mypy
* uodate
* merge main
* fix UT
2022-05-25 10:55:07 +08:00
8d3bd4dfef
Move get_max_cuda_memory and set_multi_processing to public function ( #250 )
...
* 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
2022-05-24 19:36:55 +08:00
a976257ca9
[Enhance] Support Custom LogProcessor ( #251 )
...
* support custom log processor
* supplementary docs
* format code
2022-05-24 17:17:35 +08:00
ba6b06100c
fix nested build function calls ( #244 )
2022-05-21 14:41:08 +08:00
11688507ba
[Fix] Fix some bugs in hooks and runner. ( #242 )
...
* [Fix] Fix some bugs in hooks and runner.
* fix markdown
* fix latex formula
* resolve comments
2022-05-20 17:18:24 +08:00
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
86ffc19c9c
Add pyupgrade pre-commit hook ( #232 )
...
* Add pyupgrade pre-commit hook
* fix ut
* remove comments
2022-05-19 17:56:31 +08:00
98c85529b1
[Refactor] Replace torch distributed with mmengine dist module ( #196 )
...
* [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
2022-05-19 17:40:01 +08:00
e37f1f905b
[Refactor] Make loop-related attributes to be runner's properties. ( #236 )
...
* [Enhance] Make loop related attributes to be runner's properties.
* move iter and epoch to loop
* resolve comments
2022-05-18 22:35:10 +08:00
cc8a6b86e1
[Fix] Fix BaseDataset: join prefix in parse_data_info ( #226 )
...
* implement parse_data_info
* add unit test
* fix join prefix of ann_file
* fix docstring
2022-05-17 20:53:13 +08:00
f5867f8442
Replace markdownlint with mdformat to avoid installing ruby ( #233 )
...
* Replace markdownlint with mdformat to avoid installing ruby
* fix circleci
2022-05-17 20:52:22 +08:00
fd962437e9
[Fix] Support Runner dump cfg without filename ( #228 )
...
* fix runner dump cfg
* convert dict cfg to Config
2022-05-17 17:32:10 +08:00
84e60f6463
Replace the URL of flake8 in .pre-commit-hook.yml ( #231 )
...
* pull flake8 from github
* pull flake8 from github
2022-05-17 17:30:01 +08:00
689837d2b8
[Enhancement] add pixel data and label data ( #224 )
2022-05-13 18:23:25 +08:00
63ea47460c
[Refactor] Change Runner.build_dataloader to a static method ( #219 )
...
* [Refactor] Change some methods to static methods
* only change build_dataloader to static method
2022-05-12 18:36:44 +08:00
1912660db9
[Feature] Support convert epoch-based schedulers to iter-based. ( #221 )
...
* [Feature] Support convert epoch-based schedulers to iter-based.
* Support convert and refactor LR and Momentum to mixin.
* Add unit tests
* fix args and add runner ut
* resolve comments
2022-05-10 15:17:51 +08:00
92b94e8e60
[Docs] Add docs for custom dataset wrapper ( #222 )
...
* add docs for custom dataset wrapper
* Update basedataset.md
2022-05-10 15:14:39 +08:00
22d3b04575
fix collections.abc compatibility ( #227 )
2022-05-10 15:08:51 +08:00
9087956f70
[Enhance] Update config with newest mmcv and show custom imports error explicitly ( #192 )
...
* add import error information
* Update config with newest mmcv
* add empty line to test config
2022-05-06 22:59:54 +08:00
859f4d1580
[Fix] Fix some code and docstring in LogProcessor and LoggerHook ( #213 )
...
* fix logger hook
* fix to record multi-level logs, log train/a/b to a/b
* fix loggerhook json path
* fix uunit test
* change runner.train_dataloader to runner.train_loop.dataloader
* clean debug code
* refie comments and docstring
* fix unit test
2022-05-06 14:33:33 +08:00
5c5c03e648
[Enchance] cat empty instancedata, support torch.bool for more type ( #209 )
...
* refactor instancedata
* fix docs
* fix comment
2022-05-06 14:00:51 +08:00
16058fdb18
[Fix] Fix RecursionError when key is invalid ( #220 )
...
* [Fix] Fix RecursionError when key is invalid
* refine comments
2022-05-05 20:09:03 +08:00
5007825619
[Fix] change CheckPointHook before_run to before train ( #214 )
...
* change CheckPointHook before_run to before train
* using tmp_path in each checkpointhook test case
2022-05-05 20:08:07 +08:00
a1adbff11e
[Enhancement] Get local_rank in init_dist_mpi from env ( #212 )
2022-05-05 19:55:54 +08:00
0c59eeab5f
[Enhance] Refine registry logging info. ( #206 )
...
* [Enhance] Refine registry logging info.
* update
* fix
2022-05-05 17:53:34 +08:00
4742544b25
[Feature] Support collect all registered module info. ( #193 )
...
* [Feature] Support collect all registered module info.
* update
* update
* add unit tests
* add to runner
* resolve comments
2022-05-05 11:59:51 +08:00
2da59bb1b9
fix build visualizer ( #217 )
2022-04-29 21:53:04 +08:00
29342c4eb2
[Fix] Fix to_dict in BaseDataElement and add metainfo property. ( #215 )
...
* [Fix] Fix to_dict in BaseDataElement and add metainfo property.
* Update mmengine/data/base_data_element.py
Co-authored-by: liukuikun <24622904+Harold-lkk@users.noreply.github.com>
Co-authored-by: liukuikun <24622904+Harold-lkk@users.noreply.github.com>
2022-04-29 10:19:19 +08:00
05fd5e81a3
[Enhance] Refine MMLogger and change save dir of MMLogger and Visualizer ( #205 )
...
* MMLogger can call get_current_instance without get_instance, change log path
* fix docstring
* fix docstring and update UT
* Fix runner
* fix docstring and lint
* fix ut below python3.8
* resolve circle import
2022-04-27 20:47:14 +08:00
2475c7a053
Add docstrings ( #204 )
2022-04-27 19:53:47 +08:00
b1a275a14a
improve the coverage of docstring ( #203 )
...
* improve the coverage of docstring
* update
2022-04-27 19:53:06 +08:00
db2b45b7ac
[Fix] Fix the probable failure of the basedataset unit test ( #210 )
2022-04-27 19:48:28 +08:00
8668c14d25
remove ann_file arg in load_ann_list ( #211 )
2022-04-27 19:47:49 +08:00
17dbac1812
[Enhancement] Handle the device type of inputs in functions ( #137 )
...
* [Enhancement] Handle the device type of inputs in functions
* rename and move three fucntions to dist/utils.py
* minor refinement
* rename dist to torch_dist in utils.py
* update unit tests
* refine unit tests
* add unit tests
* fix unit tests
* replace Sequence with list and tuple
* rename get_backend_device to get_comm_device
* fix unit tests
* fix unit tests
* refactor and add more unit tests
* cast_data_device does not support set type
2022-04-27 19:46:13 +08:00
661e759063
[Fix] param_scheduler can not None when training models ( #208 )
...
* [Fix] param_scheduler can not None when training models
* update unit tests
* fix unit tests
* refactor ParamSchedulerHook
* refactor unit tests
* param_schedulers can be an empty list
2022-04-27 19:45:27 +08:00
RangiLyu
Alex Yang
Mashiro
Zaida Zhou
Jiazhen Wang
jbwang1997
Jingwei Zhang
Haian Huang(深度眸)
liukuikun
Tao Gong
Yining Li