You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
pytorch-image-models/timm/models/layers
Ross Wightman 5e6dbbaf30
Add CBAM for experimentation
5 years ago
..
__init__.py Layer refactoring continues, ResNet downsample rewrite for proper dilation in 3x3 and avg_pool cases 5 years ago
activations.py Re-enable mem-efficient/jit activations after torchscript tests 5 years ago
adaptive_avgmax_pool.py Big move, layer modules and fn to timm/models/layers 5 years ago
avg_pool2d_same.py Layer refactoring continues, ResNet downsample rewrite for proper dilation in 3x3 and avg_pool cases 5 years ago
cbam.py Add CBAM for experimentation 5 years ago
cond_conv2d.py Layer refactoring continues, ResNet downsample rewrite for proper dilation in 3x3 and avg_pool cases 5 years ago
conv2d_same.py Layer refactoring continues, ResNet downsample rewrite for proper dilation in 3x3 and avg_pool cases 5 years ago
conv_bn_act.py Layer refactoring continues, ResNet downsample rewrite for proper dilation in 3x3 and avg_pool cases 5 years ago
create_attn.py Add CBAM for experimentation 5 years ago
create_conv2d.py Layer refactoring continues, ResNet downsample rewrite for proper dilation in 3x3 and avg_pool cases 5 years ago
drop.py Layer refactoring continues, ResNet downsample rewrite for proper dilation in 3x3 and avg_pool cases 5 years ago
eca.py Remove debug print from ECA module 5 years ago
helpers.py Layer refactoring continues, ResNet downsample rewrite for proper dilation in 3x3 and avg_pool cases 5 years ago
median_pool.py Big move, layer modules and fn to timm/models/layers 5 years ago
mixed_conv2d.py A bunch more layer reorg, splitting many layers into own files. Improve torchscript compatibility. 5 years ago
padding.py Layer refactoring continues, ResNet downsample rewrite for proper dilation in 3x3 and avg_pool cases 5 years ago
se.py Layer refactoring continues, ResNet downsample rewrite for proper dilation in 3x3 and avg_pool cases 5 years ago
selective_kernel.py A bunch more layer reorg, splitting many layers into own files. Improve torchscript compatibility. 5 years ago
split_batchnorm.py Big move, layer modules and fn to timm/models/layers 5 years ago
test_time_pool.py Layer refactoring continues, ResNet downsample rewrite for proper dilation in 3x3 and avg_pool cases 5 years ago