mirrors
/
mmcv
mirror of https://github.com/open-mmlab/mmcv.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
1,165 Commits
13 Branches
108 Tags
42 MiB
Commit Graph

82 Commits (f5425ab7611ab2376ddb478b57cb2f46f6054e13)

Author SHA1 Message Date
Miao Zheng 967d9d5812
[Refactoring]Add Sequential with init_weight (#884) 
* [Refactoring]Add BaseSequtial with init_weight

* revise according to comments

* revise comments

* minors

* baseseq2seq

* add modulelist

* revise minors

* fix isort

* format
 2021-03-17 15:45:28 +08:00
Ycr e076c8b042
[Feature] Support revise_keys in load_checkpoint(). (#829) 
* Simplified the code.

* Improved chkpt compatibility.

* One may modify the checkpoint via adding keywords.

* Tiny.

* Following reviewer's suggestion.

* Added unit_test.

* Fixed.

* Modify the state_dict  with  construction.

* Added test.

* Modified。

* Mimimalised the modification.

* Added the docstring.

* Format.

* Improved.

* Tiny.

* Temp file.

* Added assertion.

* Doc string.

* Fixed.
 2021-03-03 10:59:11 +08:00
Miao Zheng 72e13e6a0b
fix baserunner bug (#840) 
* fix baserunner bug

* add unit test

* hooks type check
 2021-02-20 12:46:40 +08:00
Miao Zheng a4c3702628
add initializers and BaseModule for unified parameter initialization (#780) 
* add initializers and BaseModule for unified parameter initialization

* fix circle import

* bug fix

* add is_init flag in BaseModule

* fix docstring

* sort import and fix doc format

* fix bug

* fix docformat and double quote string

* fix import sort

* import sort

* sort import

* revise according to comments

* fix doc format

* revise according to comments

* revise import and fix typo

* polish code

* revise minors

* revice minors

* revise apply function

* revise bias initialization with probability

* add type test for bias_prob

* revise minors
 2021-02-07 11:13:57 +08:00
Haian Huang(深度眸) 96892bdc42
Refactor _load_checkpoint fn (#790) 
* Refactor _load_checkpoint fn

* Update _load_checkpoint fn

* Update docs str and add unit test

* Fix unit test

* Fix lint

* Add comment and Optimize function

* Fix docs str

* Update load_ckpt and fix doc str

* Update doc str and add sort unit test

* Update and fix unit test

* Fix unit test

* Update and add unit test

* Fix openmmlab prefix error
 2021-01-28 15:09:19 +08:00
BigBigDream 8e3a801596
fix mmcv ci for parrots (#782) 
* fix mmcv ci for parrots

* fix mmcv ci

* fix lint
 2021-01-14 14:31:57 +08:00
Qiaofei Li 3d6032c6db
Add dataset classes name info to meta for saving ckpt (#776) 
* add CLASSES to meta info

* Update checkpoint.py

* add unit test for CLASSES name

* clean up the tmp folder

* use tempfile to clean up temp folder
 2021-01-08 16:40:43 +08:00
LXXXXR 09661d9754
[Feature] support upload and download checkpoint from pavimodelcloud (#725) 
* support upload and download checkpoint from pavimodelcloud

* try import pavi

* add unittest
 2020-12-22 12:59:07 +08:00
Jerry Jiarui XU 987cb58341
[Enhance] Refactor logger (#659) 
* [Enhance] Refactor logger

* fixed test

* make commit optional

* remove debug info

* fixed test
 2020-11-23 10:50:18 +08:00
Jerry Jiarui XU bb06f354d5
[Feature] Add hook msg for checkpoint hook (#635) 
* [Feature] Add hook msg for ckpt_hook

* [Feature] Add hook msg for ckpt_hook

* add test
 2020-11-06 19:56:50 +08:00
Kuro Latency bcf85026c3
Update lr_updater.py (#574) 
* Update lr_updater.py

since epoch/iteration in  runner starts with 0, we shouldn't leave the latter iteration to former (12th epoch for example, with first period equal to 12) period.

* Update lr_updater.py

* Update test_hooks.py
 2020-11-05 22:20:37 +08:00
David de la Iglesia Castro 2e6c8ec803
Refactor logger hooks (#605) 
* Refactor tags for consistency

* Fix missing runner

* Fix missing runner

* Fix missing runner

* Fix missing runner

* Fix momentum runner hook inner iter

* Fix tests

* pre-commit run
 2020-10-28 12:43:50 +08:00
Wenwei Zhang ed2887bb72
Support to specify LR of DCN's conv_offset (#344) 
* Support to specify LR of DCN's conv_offset

* Resolve comments & add unit test

* Resolve formats

* Fix CI for DCN

* Mock DCN when cpu only

* Use mock for cpu testing

* Fix docstring and support ModulatedDCN

* set offset_lr_mult as dcn's arguments, link CU-49u01p

* fix lr bug

* fall back to set LR in constructor

* resolve comments
 2020-09-25 20:51:58 +08:00
David de la Iglesia Castro 6b52e9b55f
Add runner builder (#570) 
* Add build_runner

* Parametrize test_runner

* Add imports to runner __init__

* Refactor max_iters and max_epochs from run to init

* Add assertion error messages

* Add test_builder

* Make change retro-compatible

* Raise ValueError if max_epochs and max_iters
 2020-09-25 10:25:29 +08:00
Kai Chen 66a38c86f8
Move unit tests to specific folders (#520) 
* move unit tests to specific folders

* fix path error

* remove some assertions

* fix ignore path
 2020-08-25 20:11:56 +08:00
Wang Xinjiang eacaf475f7
fix some pavi logger hooks (#481) 
* fix some pavi logger hooks

* fix unittest

* fix small bugs

* small change

* fix unittest

* Add EpochBasedRunner conditions

* Add session text

* fix small bug

* fetch runner mode from log buffer

* Add max_iter to pavi session text

* change yaml.dump to yamp.dump(yaml.load(mmcv.dump))

* Directly use by_epoch

* fix unittest

* add comments

* Use runner.epoch + 1 in pavi log

* fix runner.epoch issue for runner.mode=='val'

* fix runner.epoch issue for runner.mode=='val'

* Use abspath instead of realpath

* Add meta dump unittest

* small change

* Add comments
 2020-08-16 01:20:08 +08:00
shilong 1830347f8b
Ema (#421) 
* add ema hook

* add ema hook resume

* add ema hook test

* fix typo

* fix according to comment

* delete logger

* fix according to comment

* fix unitest

* fix typo

* fix according to comment

* change to resume_from

* typo

* fix isort
 2020-07-30 22:06:19 +08:00
Wang Xinjiang d4da3daa7e
Syncbuf (#447) 
* More robust sync buffer hook

* More robust sync buffer hook

* Reformat
 2020-07-25 12:51:46 +08:00
Wang Xinjiang 66604e83de
Add syncbuffer hook (#443) 
* reformat

* reformat

* Add register hook from cfg

* docstring

* change according to comments
 2020-07-24 14:15:44 +08:00
Jiamin 55fadb4c4e
Add runner.meta to checkpoint in save_checkpoint() (#438) 
* fix: error when runner.meta is None

* tests: add unittest for epoch-based save_checkpoint
 2020-07-20 11:40:04 +08:00
Yawei Li 7730a79fcd
fix typo of annealing (#433) 2020-07-17 23:48:22 +08:00
Harry 5704613e28
Remove all module wrapper's module when saving checkpoint (#399) 
* fix: remove all module wrapper when saving checkpoint

* refactor: move position of if

* docs: add docstring

* refactor: add _save_to_state_dict from official torch

* docs: modify docstring of _save_to_state_dict

* docs: modify docstring

* feat: add unittest

* feat: add DataParallel to unittest

* fix: a bug when model has batchnorm

* docs: update docstring
 2020-07-08 23:20:22 +08:00
Kai Chen 63b7aa31b6
Fix docstring formats (#383) 
* update doc formats

* update docstring
 2020-07-04 00:55:25 +08:00
Jintao Lin 1ebd7ea6fb
add unittest for set_random_seed (#376) 2020-07-02 00:13:04 +08:00
Harry 69048ff056
Specifying distributed training port in os.environ when training with slurm (#362) 
* feat: support for os.environ port for slurm training

* fix: port data type

* feat: add flawed unittest

* feat: add flawed unittest

* docs: add comments

* fix: unittest

* fix: unittest
 2020-06-20 00:49:44 +08:00
Kai Chen 6bb244f255
add train_step() and val_step() for MMDP (#354) 2020-06-18 20:55:53 +08:00
Harry f28a7c7ed7
Add CosineRestartLrUpdaterHook (#319) 
* feat: add CosineRestartLrUpdaterHook

* style: rename period to periods

* fix: bug in period 0

* feat: rename eta_min to min_lr and add min_lr_ratio

* docs: fix docstring of restart lr updater

* refactor: use annealing_cos

* docs: add docstring to annealing_cos

* feat: cosine restart lr update hook

* refactor: modify code order for unittest
 2020-06-15 23:01:26 +08:00
Harry 67a26da917
Add IterBasedRunner (#314) 
* feat: add IterBasedRunner

* fix: unittest

* feat: more unittest

* fix: expose dataloader len

* minor updates of BaseRunner

* refactor: remove CosineRestartLrUpdaterHook

* style: add docstring

* refactor: update IterTextLoggerHook: fstring and exp_name

* fix: epoch_runner unittest

* refactor: remove IterBasedTextLogger

* fix: old IterTextLoggerHook issue

* refactor: remove __len__ of IterLoader

* feat: add IterBasedRunner to init

* feat: add __len__ to IterLoader

* fix some docstrings

* refactor: use is_parallel_module

* fix: import issue

* fix: runner unittest missing logger

* fix checkpoints

* feat: add by_epoch default value to IterBaseRunner regitering loggger_hook

* refactor: remove setting by_epoch in log_config

* minor refactoring

* docs: add docstring

* fix: remove unused doc

* update the log info for saving checkpoints

Co-authored-by: Kai Chen <chenkaidev@gmail.com>
 2020-06-11 13:35:34 +08:00
Kai Chen 821b3ad622
Fix the BC issue of ddp (#325) 
* fix the BC issue of ddp

* minor fix for the docstring
 2020-06-08 22:34:19 +08:00
Kai Chen 35ba152821
Add a BaseRunner and rename Runner to EpochBasedRunner (#290) 
* add a BaseRunner and rename Runner to EpochBasedRunner

* fix the train/val step

* bug fix

* update unit tests

* fix unit tests

* raise an error if both batch_processor and train_step are set

* add a unit test
 2020-06-02 22:23:21 +08:00
Wenwei Zhang 19e4a06cbc
Fix CosineAnealingLr register bug (#265) 
* Fall back to CosineLr

* Fix consineanealing with unittest

* Cover momentum hook

* Add comments to explain
 2020-05-04 00:38:55 +08:00
Kai Chen a338d43d78
Refactor unittests (#241) 
* refactor unittests

* split test_video.py to two files
 2020-04-26 22:54:27 +08:00