58 Commits

Author	SHA1	Message	Date
Jintao Lin	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
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