Commit Graph

7 Commits (91e2b33d724b69f6904b2d2460cdcd7c3cdff5d0)

Author SHA1 Message Date
Ross Wightman 5e6dbbaf30 Add CBAM for experimentation
4 years ago
Ross Wightman d725991870 Remove debug print from ECA module
4 years ago
Ross Wightman 2a7d256fd5 Re-enable mem-efficient/jit activations after torchscript tests
4 years ago
Ross Wightman f902bcd54c Layer refactoring continues, ResNet downsample rewrite for proper dilation in 3x3 and avg_pool cases
4 years ago
Ross Wightman a99ec4e7d1 A bunch more layer reorg, splitting many layers into own files. Improve torchscript compatibility.
4 years ago
Ross Wightman 13746a33fc Big move, layer modules and fn to timm/models/layers
4 years ago
Ross Wightman 4defbbbaa8 Fix module name mistake, start layers sub-package
4 years ago