c1de4c9bbf
[Feature] Add spconv ops from mmdet3d ( #1581 )
...
* add ops (spconv) of mmdet3d
* fix typo
* refactor code
* resolve comments in #1452
* fix compile error
* fix bugs
* fix bug
* transform from 'types.h' to 'extension.h'
* fix bug
* transform from 'types.h' to 'extension.h' in parrots
* add extension.h in pybind.cpp
* add unittest
* Recover code
* (1) Remove prettyprint.h
(2) Switch `T` to `scalar_t`
(3) Remove useless lines
(4) Refine example in docstring of sparse_modules.py
* (1) rename from `cu.h` to `cuh`
(2) remove useless files
(3) move cpu files to `pytorch/cpu`
* reorganize files
* Add docstring for sparse_functional.py
* use dispatcher
* remove template
* use dispatch in cuda ops
* resolve Segmentation fault
* remove useless files
* fix lint
* fix lint
* fix lint
* fix unittest in test_build_layers.py
* add tensorview into include_dirs when compiling
* recover all deleted files
* fix lint and comments
* recover setup.py
* replace tv::GPU as tv::TorchGPU & support device guard
* fix lint
Co-authored-by: hdc <hudingchang.vendor@sensetime.com>
Co-authored-by: grimoire <yaoqian@sensetime.com>
2022-02-18 15:53:10 +08:00
826d3a7b68
[Bug] Avoid recursion error when using customized norm layer without _abbr_ ( #732 )
...
* avoid recursion error when using customized norm layer without _abbr_
* fix unittest bug
2020-12-22 10:19:52 +08:00
cd96a84add
[Feature] Add clamp activation layer. ( #685 )
...
* add clamp without unittest
* add clamp-act with unit test
* fix name bug
* use logical and
* fix logical_and
* fix linting
* rename ClampLayer to Clamp
* rename ClampLayer to Clamp
Co-authored-by: nbei <631557085@qq.com>
2020-12-09 10:28:32 +08:00
acee61d7c5
register deconv in CONV_LAYERS ( #582 )
...
* register deconv in CONV_LAYERS
* use ConvTranspose2d implemented in MMCV
* remove repetitive register_module
* update
* add unittest for deconv
2020-09-27 17:51:00 +08:00
7cfc839ea5
Add MMSyncBN in registry ( #420 )
...
* Add MMSyncBN in registery
* skip mmsyncbn test
2020-07-15 19:23:09 +08:00
42f03d84de
Refactor plugins: move from ops to cnn ( #380 )
...
* Refator plugins: move from ops to cnn
* minior update
* minior update
2020-07-03 01:37:58 +08:00
c35c228cbb
Support to pass args to cnn bricks ( #277 )
2020-05-12 10:41:11 +08:00
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
45111e193d
Add building bricks of cnn ( #247 )
...
* add building bricks of cnn
* add unit tests
* use registry for building bricks
* minor updates
* add scale layer
* add test for scale
* add doc string
Co-authored-by: Jiarui XU <xvjiarui0826@gmail.com>
2020-05-01 00:32:25 +08:00
Wenhao Wu
Rui Xu
GT9505
Jerry Jiarui XU
Wenwei Zhang
Kai Chen