6 Commits

Author SHA1 Message Date
Ross Wightman
ade1ba5fe3 Bring in code that should fix 1d circular padding properly, tweaks to ECA impl, using CECA in MobileNetV3 experiment 2020-02-25 16:12:41 -08:00
Ross Wightman
67e759f710 Adding support to EfficientNet / MobileNetV3 to for different attention layers in .se position 2020-02-24 18:07:09 -08:00
Ross Wightman
569419b38d Tweak some comments, add SKNet models with weights to sotabench, remove an unused branch 2020-02-15 21:18:25 -08:00
Ross Wightman
d725991870 Remove debug print from ECA module 2020-02-10 16:21:33 -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
4defbbbaa8 Fix module name mistake, start layers sub-package 2020-02-09 12:44:26 -08:00