Commit Graph

157 Commits (c9f208f7f8d5917f9a900ef2f96d9aa2ffb9264e)

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