mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity
926 Commits 141 Branches 65 Tags
Commit Graph

6 Commits

Author SHA1 Message Date
Ross Wightman
110a7c4982 AdaptiveAvgPool2d -> mean((2,3)) for all SE/attn layers to avoid NaN with AMP + channels_last layout. See https://github.com/pytorch/pytorch/issues/43992 2020-09-01 16:05:32 -07:00
Sepehr Sameni
20290b5b56
fix typo in eca 2020-04-14 15:20:35 +04:30
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