Commit Graph

333 Commits (c30e91db87bed113533dd6d0249b1f8cf09eb4cc)
 

Author SHA1 Message Date
Cao Yuhang c30e91db87
share torch version (#343) 2020-06-14 21:09:48 +08:00
lizz 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
Jintao Lin 5a98f7e421
enalbe memcached support in PetrelBackend (#341) 2020-06-13 18:32:48 +08:00
lizz 5f611bbb39
Update lr_updater.py (#334) 2020-06-12 23:40:09 +08:00
lizz 6988de82b4
Update optflow.py (#335) 2020-06-12 23:36:09 +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
Jerry Jiarui XU 61f9e91c9f
Add syntax check in .py config (#330)
* Add syntax check in .py config

* rename validate
2020-06-10 23:07:59 +08:00
Jerry Jiarui XU 2aa7712cb7
add more act w.o. inplace opt (#329) 2020-06-10 15:42:09 +08:00
Harry f5be29b2b0
fix: fix constructor docstring (#328) 2020-06-09 09:24:19 +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
Wang Xinjiang 41ca64cb4b
add gradient norm to logger (#326) 2020-06-08 17:07:59 +08:00
Joanna e7a446ae0f
add len to DC (#320)
Co-authored-by: lixuanyi <lixuanyi@sensetime.com>
2020-06-05 15:02:11 +08:00
wang shiguang 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
Ye Liu 1cb3e36a1e
Fix typo (#318) 2020-06-05 14:03:19 +08:00
Harry 6875dcfe48
refactor: use pat wrapper in default_constuctor (#317) 2020-06-04 19:31:45 +08:00
Jerry Jiarui XU a3ae781789
add reserved key in config (#315) 2020-06-04 14:55:28 +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
Harry 6f21d8b560
Add optimizer constructor from mmdetection (#313)
* feat: add optimizer constructor

* refactor: version
2020-06-02 19:57:24 +08:00
Miguel Mendez fe4f657f03
Fix deprecation warning collections.abc (#312)
* Fix deprecation warning collections.abc

* Fix isort
2020-06-02 19:29:54 +08:00
Harry 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
Kai Chen 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
Kai Chen 4150d0ac2f
Update build.yml 2020-05-30 13:09:09 +08:00
Jerry Jiarui XU 5542c24bb1
Fixed package data in pip (#303) 2020-05-30 13:04:37 +08:00
Wenwei Zhang 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
Wang Xinjiang b63d880af5
Add sleep(2) during epoch transition (#306) 2020-05-30 13:00:01 +08:00
Wenwei Zhang 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
Jerry Jiarui XU 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
Jerry Jiarui XU 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
Kai Chen c63e4d57c3
bump version to 0.5.6 (#298) 2020-05-26 00:39:14 +08:00
Miguel Mendez fa470ebcef
Force copy of input img (#297) 2020-05-24 17:45:21 +08:00
Wenwei Zhang 02f2573a29
Add regnetx links (#287) 2020-05-23 15:34:46 +08:00
Jerry Jiarui XU 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
Kai Chen 3bf3e8ef24
Revert "add PetrelBackend (#291)" (#293)
This reverts commit 7f80a2c6b9.
2020-05-23 01:07:43 +08:00
Jerry Jiarui XU 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
Jerry Jiarui XU 6ce22a13e6
Add res2net pretrain model (#289)
* add res2net and regnetx

* remove regnetx
2020-05-21 19:05:42 +08:00
Kai Chen 6004776241
fix the install_requires (#288) 2020-05-19 14:41:59 +08:00
Jerry Jiarui XU fabee855b1
Add requirement parser (#286) 2020-05-15 23:08:12 +08:00
Lei Yang eb93599e1b
Add yapf to install_requires (#285) 2020-05-15 20:25:59 +08:00
Kai Chen 27e3284e26
bump the version to 0.5.3 (#284) 2020-05-15 01:51:19 +08:00
Kai Chen feb650642e
Bug fix for the deploy workflow (#283)
* update the deploy workflow

* fix a typo

* update MANIFEST.in to include pyx

* remove debug info
2020-05-14 23:00:22 +08:00
Kai Chen 506df8cdce
Add a pypi workflow and bump version to 0.5.2 (#282)
* add a workflow to publish to pypi and remove travis.yml

* move the if-statement to the job level

* add a badge

* switch to ubuntu-latest

* bump version to 0.5.2

* add apt-get update before installing
2020-05-14 16:41:40 +08:00
Jerry Jiarui XU 83bcd1134d
Add yapf for Config.pretty_text (#281)
* add yapf for pretty text

* add yapf style
2020-05-14 14:17:39 +08:00
wang shiguang a805cd9d91
wrapper extension (#279)
* wrapper extension

* fix

Co-authored-by: wangshiguang <wangshiguang@sensetime.com>
2020-05-14 14:15:58 +08:00
Kai Chen 693f61af76
Add GitHub action flow (#280)
* Create a github action workflow

* Fix the workflow config

* Fix codecov

* skip codecov for python 3.8

* fix the expression

* remove the if expression
2020-05-14 10:26:27 +08:00
Jintao Lin 25d07e3d54
Support conv layers' own `init_weights` method (#278)
* Support conv layers' own `init_weights` method

* Add related unittest about ConvModule init_weight

* make the comments more specific
2020-05-13 22:58:16 +08:00
Wenwei Zhang c35c228cbb
Support to pass args to cnn bricks (#277) 2020-05-12 10:41:11 +08:00
Jerry Jiarui XU d5936d02e3
Make Config.dump() output in original file (#275)
* add dump() from for different file type

* remove exception
2020-05-11 21:48:56 +08:00
Kai Chen d4fac3a6dd
Use parrots wrapper (#272)
* use parrots wrapper for norms

* add unittests for is_norm()

* add a test case
2020-05-08 11:24:35 +08:00
Kai Chen 6aa5131559
move test_weight_init to tests/test_cnn (#270) 2020-05-05 21:03:12 +08:00
Kai Chen d2e3a84315
add missing requirements for the doc building (#269) 2020-05-05 14:22:23 +08:00