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.
Ross Wightman
c60069c1eb
Annotate types on drop fns to avoid torchscript error
|
5 years ago |
.. |
__init__.py
|
Update comments for Selective Kernel and DropBlock/Path impl, add skresnet34 weights
|
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
|
Tweak some comments, add SKNet models with weights to sotabench, remove an unused branch
|
5 years ago |
cond_conv2d.py
|
Tweak some comments, add SKNet models with weights to sotabench, remove an unused branch
|
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
|
Annotate types on drop fns to avoid torchscript error
|
5 years ago |
eca.py
|
Tweak some comments, add SKNet models with weights to sotabench, remove an unused branch
|
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
|
Tweak some comments, add SKNet models with weights to sotabench, remove an unused branch
|
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
|
Update comments for Selective Kernel and DropBlock/Path impl, add skresnet34 weights
|
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 |