louzana
2c6fc5fd9b
fix bug of building ConvModule with HSigmoid using inplace=True and a… ( #369 )
...
* fix bug of building ConvModule with HSigmoid using inplace=True and add corresponding unittest
* fix linting
2020-06-28 23:27:14 +08:00
Jerry Jiarui XU
2aa7712cb7
add more act w.o. inplace opt ( #329 )
2020-06-10 15:42:09 +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
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
Kai Chen
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