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
Wenwei Zhang
c35c228cbb
Support to pass args to cnn bricks ( #277 )
2020-05-12 10:41:11 +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
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
