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
0004f37d25
Some TResNet cleanup.
...
* allow use of global pool arg, test-time-pooling
* clean checkpoints to just contain state dict, add 448 res checkpoints
* support DataParallel via lazy filter creation for JIT Downsample
* some minor formatting (mostly alignment) preferences
5 years ago
..
__init__.py
TResNet models
5 years ago
activations.py
Re-enable mem-efficient/jit activations after torchscript tests
5 years ago
adaptive_avgmax_pool.py
Some TResNet cleanup.
5 years ago
anti_aliasing.py
Some TResNet cleanup.
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
Update EfficientNet feature extraction for EfficientDet. Add needed MaxPoolSame as well.
5 years ago
conv2d_same.py
Update EfficientNet feature extraction for EfficientDet. Add needed MaxPoolSame as well.
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
Fix minor typos in create_attn.py and resnet.py
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
Update EfficientNet feature extraction for EfficientDet. Add needed MaxPoolSame as well.
5 years ago
pool2d_same.py
Update EfficientNet feature extraction for EfficientDet. Add needed MaxPoolSame as well.
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
space_to_depth.py
Some TResNet cleanup.
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