mirrors
/
mmengine
mirror of https://github.com/open-mmlab/mmengine.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
314 Commits
31 Branches
27 Tags
20 MiB
Commit Graph

314 Commits (a706bbc0186c39df5d4d5da58486acb3b91a3541)
 

Author SHA1 Message Date
Mashiro a706bbc018
[Fix]: fix error and add unit test (#429) 2022-08-17 19:16:00 +08:00
Zaida Zhou f5cb45dc33
[Doc]: fix typo in installation command (#433) 2022-08-16 15:42:27 +08:00
Mashiro 381aa429df
[Fix] Make default scope of runner be `mmengine` (#435) 
* make default scope of runner be mmengine

* fix as comment
 2022-08-16 15:41:57 +08:00
Mashiro ac8686e28b
[Fix] Make logger compatible with python3.6 (#437) 
* Makr logger compatible with python3.6

* minor refine

* fix as comment
 2022-08-16 15:38:52 +08:00
Zaida Zhou 7b3397c614
Add .pre-commit-config-zh-cn.yaml (#405) 2022-08-15 11:50:53 +08:00
Zaida Zhou 26342f3586
[Docs] Add the steps to build docs (#409) 2022-08-15 11:47:08 +08:00
Kai Hu 3b1c45bb76
[Fix] fix the docstring to match the order of the function return (#415) 2022-08-15 11:46:19 +08:00
Zaida Zhou 4637c49640
[Refactor] Rename test_hook to test_hooks (#425) 2022-08-15 11:45:07 +08:00
Zaida Zhou 8d620e59fb
Directly import from collections.abc (#427) 2022-08-15 11:43:42 +08:00
Mashiro 438e8e7498
BaseModel support recursively set the device of data_preprocessor (#387) 2022-08-15 11:04:24 +08:00
Zaida Zhou f98ba60629
[Enhancement] Improve unit tests of mmengine/runner (#182) 
* [Enhancement] Add unit test for get_priority

* fix priority ut

* fix typo

Co-authored-by: Wenwei Zhang <40779233+ZwwWayne@users.noreply.github.com>
 2022-08-15 10:57:58 +08:00
Mashiro 2708b7ed48
fix ci (#424) 2022-08-13 09:15:08 +08:00
RangiLyu 813f49bf23
[Feature] Support CosineRestartParamScheduler. (#397) 
* [Feature] Support CosineRestartParamScheduler.

* add ut and docstring

* add docstring
 2022-08-11 17:57:35 +08:00
Mashiro b14cbc2576
[Fix] Fix wrong epoch and iter when saving best ckpt (#400) 
* fix wrong epoch andd iter when save bbest ckpt

* fix ut

* fix resume best ckpt unexpectedly

* minor refine

* fix unit test
 2022-08-11 14:52:38 +08:00
Mashiro ee56f151f6
[Fix] Support training with data without `metainfo`. (#417) 
* support training with data without metainfo

* clean the code

* clean the code
 2022-08-11 14:51:11 +08:00
Mashiro c287e1fb92
Fix import mmsegmentation.utils error (#414) 2022-08-09 16:20:46 +08:00
Ma Zerun 9b2a0e02da
[Enhance] Add `data_preprocessor` config as an argument of runner. (#343) 
* [Enhance] Add `preprocess_cfg` as an argument of runner.

* Rename `preprocess_cfg` to `data_preprocessor`

* Fix docstring
 2022-08-09 11:25:29 +08:00
Mashiro cd257c9047
[Enhance] Support log enviroment information during initiate runner (#384) 
* Support log enviroment information when initiate runner

* Fix unit test

* fix as comment, save world_size

* log gpu num

* clear code and reformat log

* minor refine

* fix as comment

* minor refine

* clean the code

* clean the code

* remove save world_size in meta
 2022-08-08 22:23:52 +08:00
Mashiro 7e2302388f
[Feature] add config new feature (#105) 2022-08-08 21:01:06 +08:00
Mashiro a07a063306
[Enhance] Add build function for scheduler. (#372) 
* add build function for scheduler

* add unit test

add unit test

* handle convert_to_iter in build_scheduler_from_cfg

* restore deleted code

* format import

* fix lint
 2022-08-08 20:34:16 +08:00
Mashiro 99de0951af
[Enhance] EMAHook support does not load checkpoint strictly (#352) 
* BaseAveragedModel support load ckpt without module prefix

* refine docstring

* allow EMAHook does not load ckpt strictly

* add unit test for strict argument of EMAHook

* sync remote

* sync remote

* clean the code

* ema hook supports setting start iter

* fix unit test

* fix as comment

* fix as comment

* describe kwargs
 2022-08-08 20:26:16 +08:00
LeoXing1996 08602a2385
[Enhancement] Support save best based on multi metrics (#349) 
* support save best based on multi metrics

* add unit test

* resolve bugs after rebasing

* revise docstring

* revise docstring

* fix as comment

* revise as comment
 2022-08-08 20:17:17 +08:00
Mashiro 6ebb7ed481
[Fix] Fix config cannot change(assign) the value of the dict in config(without base config) (#378) 
* Support changing value of dict in config(without base config)

* minor refine
 2022-08-08 17:07:47 +08:00
Mashiro 5580542666
[Fix] Fix build multiple list of scheduler for multiple optimizers (#383) 
* fix build multiple scheduler

* add new unit test

* fix comment and error message

* fix comment and error message

* extract _parse_scheduler_cfg

* always call build_param_scheduler during train and resume. If there is only one optimizer, the defaut value for sheduler will be a list, otherwise there is multiple optimizer, the default value of sheduler will be a dict

* minor refine

* rename runner test exp name

* fix as comment

* minor refine

* fix ut

* only check parameter scheduler

* minor refine
 2022-08-08 17:05:27 +08:00
Mashiro 1a8f013937
[Refine] Make scheduler default to None (#396) 
* make scheduler default to None

* fix bc breaking

* refine warning message

* fix as comment

* fix as comment

* fix lint
 2022-08-04 20:13:13 +08:00
Mashiro df4e6e3294
[Fix] Fix resume `message_hub` and save `metainfo` in message_hub. (#394) 
* Fix resume message hub and save metainfo in messagehub

* fix as comment
 2022-08-02 19:06:35 +08:00
Mashiro eb25129935
[Feature]: add epoch log scalar (#391) 2022-08-02 19:01:06 +08:00
Mashiro f850de71c3
[Fix] Support use 'global variable' in config function (#390) 
* Support use 'global var' in config function

* upload test file
 2022-08-01 20:10:10 +08:00
Yuan Liu 81c3de54b9
[Fix]: Fix resume bug (#389) 
* [Fix]: Fix resume bug

* [Fix]: Change last_checkpoint check logic

* [Fix]: Fix lint

* [Fix]: Change warning to print_log
 2022-08-01 13:20:08 +08:00
RangiLyu 84cd19aaa2
[Enhance] Support convert float begin&end in ParamScheduler. (#393) 2022-08-01 10:26:58 +08:00
RangiLyu 4432e54c97
[Fix] Fix gpu tensors in results list are not on the same device. (#385) 
* [Fix] Fix gpu tensors in results list are not on the same device.

* cast all tensor to cpu
 2022-08-01 10:20:57 +08:00
RangiLyu cfee85ff16
[Enhance] Return loop results. (#392) 2022-07-30 20:22:52 +08:00
Miao Zheng 39e7efb04d
[Fix] Revise UT of OneCycle schedulor (#388) 2022-07-27 16:22:00 +08:00
Mashiro ec72f59bf3
[Enhance] Remove blink info and debug level message (#371) 
* remove blink info debug msg

* minor refine

* minor refine
 2022-07-22 14:41:51 +08:00
RangiLyu f3189918e5
[Fix] Fix logged and current weight not on the same device. (#382) 2022-07-22 14:41:11 +08:00
RangiLyu 16ef54c45d
[Fix] Fix compatibility of persistent_workers. (#379) 2022-07-21 16:02:51 +08:00
Mashiro 6bd548d8bb
[FIx] Logging time in LoggerHook.after_val_epoch has no effect (#376) 2022-07-20 16:51:11 +08:00
Mashiro 6b47035fdf
[Fix] Fix save scheduler state dict with optim wrapper (#375) 
* fix save scheduler state dict with optim wrapper

* remove for loop and inherit TestParameterScheduler

* remove for loop and inherit TestParameterScheduler

* minor refine
 2022-07-20 16:32:48 +08:00
Tong Gao 5b065b10fd
[Enhance] Support Compose(None) (#373) 
* [Enhance] Allow Compose(None)

* add typehint

* fix
 2022-07-20 16:04:24 +08:00
Mashiro ea61bf6bb7
Fix: avoid modification of scalar_dict in LocalVisBackend (#377) 2022-07-20 16:03:01 +08:00
Mashiro 5b648c119f
[Fix] Fix LogProcessor._get_iter (#370) 2022-07-19 18:30:59 +08:00
RangiLyu 1241c21296
[Fix] Fix weight initializing in test and refine registry logging. (#367) 
* [Fix] Fix weight initializing and registry logging.

* sync params

* resolve comments
 2022-07-19 18:28:57 +08:00
Ma Zerun 3da66d1f87
[Enhance] Auto set the `end` of param schedulers. (#361) 
* [Enhance] Auto set the `end` of param schedulers.

* Add log output and unit test

* Update docstring

* Update unit tests of `CosineAnnealingParamScheduler`.
 2022-07-15 19:53:28 +08:00
Mashiro 78fad67d0d
[Fix] fix resume message_hub (#353) 
* fix resume message_hub

* add unit test

* support resume from messagehub

* minor refine

* add comment

* fix typo

* update docstring
 2022-07-14 20:13:22 +08:00
Mashiro c9c6d454f1
init OptimWrapper after resume (#362) 2022-07-14 20:10:48 +08:00
Mashiro 45001a1f6f
[Enhance] Support using variables in base config directly as normal variables. (#329) 
* first commit

* Support modify base config and add unit test

* remove import mmengine in config

* add unit test

* fix lint

* add unit test

* move RemoveAssignFromAST to config utils

* git add utils

* fix format issue in test file

* refine unit test

* refine unit test
 2022-07-14 13:05:55 +08:00
Mashiro 6b608b4ef1
[Enhance] Add `build_model_from_cfg` (#328) 
* clean code

* fix as comment

* fix as comment

* add get_registry_by_scope method

* add unit test and docstring example

* rename get_registry_by_scope to switch_scope_and_registry

* move build function to registry/builder

* fix docstring

* rename builder->registry_builder, move build_from_cfg to registry_builder

rename builder->registry_builder, move build_from_cfg to registry_builder

* rename registry_builder to build_function

rename registry_builder to build_function

* fix docstring and type hint

* rename build_function to build_functions
 2022-07-13 19:01:59 +08:00
ChaimZhu acd9cab25a
[Fix] fix `is_abs` criterion error (#358) 
* fix abs criterion error

* update
 2022-07-12 11:16:08 +08:00
Mashiro 936c4ebc58
[Fix] Fix missing device ids in wrap_model (#351) 
* fix missing device ids in wrap_model

* clean the code

* use default broadcast_buffers

* refine MMSeparateDistributedDataParallel

* rename tmp variable

* refine docstring

* add type hints

* refactor docstring of ddp model

* add  arg in docstring

* minor refine

* better ddp link
 2022-07-08 15:01:47 +08:00
VVsssssk 792f481efe
[Fix]fix ClassBalancedDataset (#354) 
* fix cbds

* fix
 2022-07-08 14:51:51 +08:00