fef7b2fea4
add hswish and hsigmoid ( #358 )
...
* add hswish and hsigmoid
* fix linting
* fix linting
* delete useless staticmethod in cnn/bricks/hsigmoid.py
* forward(x) ==> forward(seld, x)
* add unittests for hsigmoid and hswish
* use torch.equal instead of numpy.equal
2020-06-19 21:35:17 +08:00
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
2aa7712cb7
add more act w.o. inplace opt ( #329 )
2020-06-10 15:42:09 +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
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
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
6aa5131559
move test_weight_init to tests/test_cnn ( #270 )
2020-05-05 21:03:12 +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
louzana
Jintao Lin
Jerry Jiarui XU
Harry
Jintao Lin
Wenwei Zhang
Kai Chen