dcc20f3a4b
Add `_NonLocalNd` module ( #331 )
...
* add non_local module
* rewrite non local module comments
* perfect docstring and adjust init function
* not to init norm layer
* Correct initialize when there is a norm
* set normal method for both embedded_gaussian and dot_product
2020-06-18 20:56:48 +08:00
6bb244f255
add train_step() and val_step() for MMDP ( #354 )
2020-06-18 20:55:53 +08:00
c74d729d92
Add custom_group to DefaultOptimizerConstrutor ( #347 )
...
* feat: add custom_group to DefaultOptimConstrutor
* refactor: move custom_groups validate to _validate_cfg
* docs: add doc to explain custom_groups
* feat: add unittest for non_exist_key
* refactor: one param per group
* fix: small fix
* fix: name
* docs: docstring
* refactor: change to mult for only lr and wd custom
* docs: docstring
* docs: more explaination
* feat: sort custom key
* docs: add docstring
* refactor: use reverse arg of sorted
* docs: fix comment
* docs: fix comment
* refactor: small modi
* refactor: small modi
* refactor: small modi
2020-06-17 23:39:50 +08:00
657f03ad08
replace DeprecationWarning with UserWarning ( #353 )
2020-06-17 20:52:48 +08:00
799421003d
Add module wrapper registry ( #352 )
...
* feat: modify parallel to module wrapper
* fix: unittest
* fix: add registry file
* feat: add module wrapper unittest
2020-06-17 20:12:54 +08:00
c6784f4acf
add backend arg ( #350 )
...
* add backend arg
* add test
* update doc, add test
* update test
* update doc
2020-06-17 19:42:31 +08:00
f48241a65a
load_checkpoint support normal dict checkpoints ( #351 )
...
* load_checkpoint support normal dict checkpoints
* comments
2020-06-17 17:10:04 +08:00
97f9efd825
fix bug in using inner iter when by_epcoh==True ( #346 )
...
* fix bug in using inner iter when by_epcoh==True
* set default for log config
* fix bug: remove optim in val, use iter in log when by_epoch is False
2020-06-17 14:01:11 +08:00
630b747cb1
added pillow backend in loading ( #333 )
...
* added pillow backend in loading
* reformat
* remove auto fill for RGBA
* change according to comments
* change according to comments
* reformat
2020-06-17 10:25:48 +08:00
3f4c9fd0c8
fix possible bugs brought by dumping grad_norm to the logger in pytorch1.5 ( #349 )
2020-06-16 23:16:26 +08:00
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
9f04477f93
Harmless changes ( #340 )
...
* harmless changes
Signed-off-by: lizz <lizz@sensetime.com>
* Update conv_module.py
* No bracket for simple class
Signed-off-by: lizz <lizz@sensetime.com>
2020-06-15 11:29:01 +08:00
e8fa66aa16
Add regnet 400mf model ( #342 )
2020-06-14 21:10:05 +08:00
c30e91db87
share torch version ( #343 )
2020-06-14 21:09:48 +08:00
b87e774f66
Remove more unused things ( #338 )
...
* Remove more unused things
Signed-off-by: lizz <lizz@sensetime.com>
* panic
Signed-off-by: lizz <lizz@sensetime.com>
2020-06-13 20:19:30 +08:00
5a98f7e421
enalbe memcached support in PetrelBackend ( #341 )
2020-06-13 18:32:48 +08:00
5f611bbb39
Update lr_updater.py ( #334 )
2020-06-12 23:40:09 +08:00
6988de82b4
Update optflow.py ( #335 )
2020-06-12 23:36:09 +08:00
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
61f9e91c9f
Add syntax check in .py config ( #330 )
...
* Add syntax check in .py config
* rename validate
2020-06-10 23:07:59 +08:00
2aa7712cb7
add more act w.o. inplace opt ( #329 )
2020-06-10 15:42:09 +08:00
f5be29b2b0
fix: fix constructor docstring ( #328 )
2020-06-09 09:24:19 +08:00
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
41ca64cb4b
add gradient norm to logger ( #326 )
2020-06-08 17:07:59 +08:00
e7a446ae0f
add len to DC ( #320 )
...
Co-authored-by: lixuanyi <lixuanyi@sensetime.com>
2020-06-05 15:02:11 +08:00
d128398121
Parrots wrapper ( #316 )
...
* add parrots and pytorch dataloader wrapper
* fix commit
Co-authored-by: wangshiguang <wangshiguang@sensetime.com>
2020-06-05 14:07:01 +08:00
1cb3e36a1e
Fix typo ( #318 )
2020-06-05 14:03:19 +08:00
6875dcfe48
refactor: use pat wrapper in default_constuctor ( #317 )
2020-06-04 19:31:45 +08:00
a3ae781789
add reserved key in config ( #315 )
2020-06-04 14:55:28 +08:00
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
6f21d8b560
Add optimizer constructor from mmdetection ( #313 )
...
* feat: add optimizer constructor
* refactor: version
2020-06-02 19:57:24 +08:00
fe4f657f03
Fix deprecation warning collections.abc ( #312 )
...
* Fix deprecation warning collections.abc
* Fix isort
2020-06-02 19:29:54 +08:00
efecf7d1fe
fix building ConvModule with Tanh activation ( #309 )
...
* fix: tanh has not inplace
* feat: add ConvModule with Tanh
2020-06-01 21:26:51 +08:00
213156cebb
Allow registering new backends with decorators ( #307 )
...
* allow registering new backends with decorators
* add a docstring
* minor update to the docstring
2020-05-31 21:56:03 +08:00
4150d0ac2f
Update build.yml
2020-05-30 13:09:09 +08:00
5542c24bb1
Fixed package data in pip ( #303 )
2020-05-30 13:04:37 +08:00
8ceb404ea6
Use mapping rather than dict for special keys ( #304 )
...
* Support path as a key in dict of config
* reformat test case
* update pre-commit version and fix format
* fix bug
* clean code
* reformat
* fix missing parts
2020-05-30 13:03:33 +08:00
b63d880af5
Add sleep(2) during epoch transition ( #306 )
2020-05-30 13:00:01 +08:00
ff89d36612
Add resnext_101_32x8d model ( #300 )
...
* add resnext_101_32x8d model
* rename file
* rm bgr
* Upgrade version
2020-05-27 23:30:16 +08:00
d6411b7fff
BC of model zoo: add deprecate urls ( #301 )
...
* add deprecate urls
* add deprecate urls
* warning test
* rename to deprecated.json
2020-05-27 22:09:06 +08:00
37d8facfad
Support default/external json for open-mmlab models ( #230 )
...
* support default/external json for open-mmlab models
* add local
* add more test
* add docs
* add docs
* update docs
* refactor
* add json in MANIFEST
* fixed json typo
2020-05-27 17:12:43 +08:00
c63e4d57c3
bump version to 0.5.6 ( #298 )
2020-05-26 00:39:14 +08:00
fa470ebcef
Force copy of input img ( #297 )
2020-05-24 17:45:21 +08:00
02f2573a29
Add regnetx links ( #287 )
2020-05-23 15:34:46 +08:00
0946feabe3
add PetrelBackend ( #294 )
...
* add PetrelBackend
* update docs
* add path_maps for Ceph
* rename to path_mapping
* fixed import
2020-05-23 01:31:11 +08:00
3bf3e8ef24
Revert "add PetrelBackend ( #291 )" ( #293 )
...
This reverts commit 7f80a2c6b9
2020-05-23 01:07:43 +08:00
7f80a2c6b9
add PetrelBackend ( #291 )
...
* add PetrelBackend
* update docs
* add path_maps for Ceph
* rename to path_mapping
2020-05-22 12:31:34 +08:00
6ce22a13e6
Add res2net pretrain model ( #289 )
...
* add res2net and regnetx
* remove regnetx
2020-05-21 19:05:42 +08:00
6004776241
fix the install_requires ( #288 )
2020-05-19 14:41:59 +08:00
fabee855b1
Add requirement parser ( #286 )
2020-05-15 23:08:12 +08:00
Jintao Lin
Kai Chen
Harry
Jerry Jiarui XU
Wang Xinjiang
Rui Xu
lizz
Wenwei Zhang
Cao Yuhang
Joanna
wang shiguang
Ye Liu
Miguel Mendez
Wang Xinjiang