5 Commits (afb6bd066910e71fb7b2621ab098a3826ccc27b8)

Author	SHA1	Message	Date
Ross Wightman	1904ed8fec	Improve dropblock impl, add fast variant, and better AMP speed, inplace, batchwise... few ResNeSt cleanups	2020-05-13 15:17:08 -07:00
Ross Wightman	c60069c1eb	Annotate types on drop fns to avoid torchscript error	2020-02-27 09:30:23 -08:00
Ross Wightman	f1d5f8a6c4	Update comments for Selective Kernel and DropBlock/Path impl, add skresnet34 weights	2020-02-18 13:58:30 -08:00
Ross Wightman	f902bcd54c	Layer refactoring continues, ResNet downsample rewrite for proper dilation in 3x3 and avg_pool cases ... * select_conv2d -> create_conv2d * added create_attn to create attention module from string/bool/module * factor padding helpers into own file, use in both conv2d_same and avg_pool2d_same * add some more test eca resnet variants * minor tweaks, naming, comments, consistency	2020-02-10 11:55:03 -08:00
Ross Wightman	a99ec4e7d1	A bunch more layer reorg, splitting many layers into own files. Improve torchscript compatibility.	2020-02-09 14:46:28 -08:00