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 0ea53cecc3 Merge branch 'master' into densenet_update_and_more		4 years ago
..
__init__.py	Merge branch 'master' into densenet_update_and_more	4 years ago
activations.py	Re-enable mem-efficient/jit activations after torchscript tests	4 years ago
adaptive_avgmax_pool.py	Some TResNet cleanup.	4 years ago
anti_aliasing.py	Final blurpool2d cleanup and add resnetblur50 weights, match tresnet Downsample arg order to BlurPool2d for interop	4 years ago
blur_pool.py	Final blurpool2d cleanup and add resnetblur50 weights, match tresnet Downsample arg order to BlurPool2d for interop	4 years ago
cbam.py	Tweak some comments, add SKNet models with weights to sotabench, remove an unused branch	4 years ago
cond_conv2d.py	Update EfficientNet feature extraction for EfficientDet. Add needed MaxPoolSame as well.	4 years ago
conv2d_same.py	Update EfficientNet feature extraction for EfficientDet. Add needed MaxPoolSame as well.	4 years ago
conv_bn_act.py	Fix #139. Broken SKResNets after BlurPool addition, as a plus, SKResNets support AA now too.	4 years ago
create_attn.py	Fix minor typos in create_attn.py and resnet.py	4 years ago
create_conv2d.py	Layer refactoring continues, ResNet downsample rewrite for proper dilation in 3x3 and avg_pool cases	4 years ago
create_norm_act.py	Add norm_act factory method, move JIT of norm layers to factory	4 years ago
drop.py	Improve dropblock impl, add fast variant, and better AMP speed, inplace, batchwise... few ResNeSt cleanups	4 years ago
eca.py	fix typo in eca	4 years ago
evo_norm.py	Add norm_act factory method, move JIT of norm layers to factory	4 years ago
helpers.py	Layer refactoring continues, ResNet downsample rewrite for proper dilation in 3x3 and avg_pool cases	4 years ago
median_pool.py	Big move, layer modules and fn to timm/models/layers	4 years ago
mixed_conv2d.py	Tweak some comments, add SKNet models with weights to sotabench, remove an unused branch	4 years ago
norm_act.py	DenseNet converted to support ABN (norm + act) modules. Experimenting with EvoNorm, IABN	4 years ago
padding.py	Update EfficientNet feature extraction for EfficientDet. Add needed MaxPoolSame as well.	4 years ago
pool2d_same.py	Update EfficientNet feature extraction for EfficientDet. Add needed MaxPoolSame as well.	4 years ago
se.py	Add RegNet models and weights	4 years ago
selective_kernel.py	Fix #139. Broken SKResNets after BlurPool addition, as a plus, SKResNets support AA now too.	4 years ago
space_to_depth.py	Remove some CRLF that made it into TResNet merge	4 years ago
split_attn.py	Improve dropblock impl, add fast variant, and better AMP speed, inplace, batchwise... few ResNeSt cleanups	4 years ago
split_batchnorm.py	Big move, layer modules and fn to timm/models/layers	4 years ago
test_time_pool.py	Layer refactoring continues, ResNet downsample rewrite for proper dilation in 3x3 and avg_pool cases	4 years ago
weight_init.py	Remove annoying InceptionV3 dependency on scipy and insanely slow trunc_norm init. Bring InceptionV3 code into this codebase and use upcoming torch trunch_norm_ init.	4 years ago