55 Commits (86f8ade9597777342cb4c24023de3ba1eeba3b8b)

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