Commit Graph

120 Commits (4027412757f01744a2d947a22c0042bb387c3266)

Author SHA1 Message Date
Sepehr Sameni 20290b5b56
fix typo in eca
4 years ago
talrid 6209146738 TResNet models
4 years ago
Ross Wightman 1a8f5900ab Update EfficientNet feature extraction for EfficientDet. Add needed MaxPoolSame as well.
4 years ago
Chris Ha 06a50a94a8 Fix minor typos in create_attn.py and resnet.py
4 years ago
Ross Wightman 1a9ab07307 One too many changes at a time, fix missing C
4 years ago
Ross Wightman f17b42bc33 Blur filter no longer a buffer
4 years ago
Ross Wightman 6cdeca24a3 Some cleanup and fixes for initial BlurPool impl. Still some testing and tweaks to go...
4 years ago
Chris Ha acd1b6cccd Implement Functional Blur on resnet.py
4 years ago
Chris Ha ce3d82b58b Update blurpool.py
4 years ago
Chris Ha 3a287a6e76 Create blurpool.py
4 years ago
Ross Wightman c60069c1eb Annotate types on drop fns to avoid torchscript error
4 years ago
Ross Wightman f1d5f8a6c4 Update comments for Selective Kernel and DropBlock/Path impl, add skresnet34 weights
4 years ago
Ross Wightman 569419b38d Tweak some comments, add SKNet models with weights to sotabench, remove an unused branch
4 years ago
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