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
f902bcd54c
Layer refactoring continues, ResNet downsample rewrite for proper dilation in 3x3 and avg_pool cases
...
* select_conv2d -> create_conv2d
* added create_attn to create attention module from string/bool/module
* factor padding helpers into own file, use in both conv2d_same and avg_pool2d_same
* add some more test eca resnet variants
* minor tweaks, naming, comments, consistency
5 years ago
..
layers
Layer refactoring continues, ResNet downsample rewrite for proper dilation in 3x3 and avg_pool cases
5 years ago
__init__.py
Big move, layer modules and fn to timm/models/layers
5 years ago
densenet.py
Big move, layer modules and fn to timm/models/layers
5 years ago
dla.py
Big move, layer modules and fn to timm/models/layers
5 years ago
dpn.py
Big move, layer modules and fn to timm/models/layers
5 years ago
efficientnet.py
Layer refactoring continues, ResNet downsample rewrite for proper dilation in 3x3 and avg_pool cases
5 years ago
efficientnet_blocks.py
Layer refactoring continues, ResNet downsample rewrite for proper dilation in 3x3 and avg_pool cases
5 years ago
efficientnet_builder.py
A bunch more layer reorg, splitting many layers into own files. Improve torchscript compatibility.
5 years ago
factory.py
Update EfficientNet comments, MobileNetV3 non-TF create fns, fix factory arg checks, bump PyTorch version req to 1.2
5 years ago
feature_hooks.py
Bring EfficientNet and MobileNetV3 up to date with my gen-efficientnet repo
5 years ago
gluon_resnet.py
Layer refactoring continues, ResNet downsample rewrite for proper dilation in 3x3 and avg_pool cases
5 years ago
gluon_xception.py
Big move, layer modules and fn to timm/models/layers
5 years ago
helpers.py
Add checkpoint averaging script. Add headers, shebangs, exec perms to all scripts
5 years ago
hrnet.py
Big move, layer modules and fn to timm/models/layers
5 years ago
inception_resnet_v2.py
Big move, layer modules and fn to timm/models/layers
5 years ago
inception_v3.py
Update version, results csv files, and move remaining dropbox weights to github
5 years ago
inception_v4.py
Big move, layer modules and fn to timm/models/layers
5 years ago
mobilenetv3.py
Layer refactoring continues, ResNet downsample rewrite for proper dilation in 3x3 and avg_pool cases
5 years ago
nasnet.py
Big move, layer modules and fn to timm/models/layers
5 years ago
pnasnet.py
Big move, layer modules and fn to timm/models/layers
5 years ago
registry.py
Model updates. Add my best ResNet50 weights top-1=78.47. Add some other torchvision weights.
5 years ago
res2net.py
Layer refactoring continues, ResNet downsample rewrite for proper dilation in 3x3 and avg_pool cases
5 years ago
resnet.py
Layer refactoring continues, ResNet downsample rewrite for proper dilation in 3x3 and avg_pool cases
5 years ago
selecsls.py
Big move, layer modules and fn to timm/models/layers
5 years ago
senet.py
Big move, layer modules and fn to timm/models/layers
5 years ago
sknet.py
Layer refactoring continues, ResNet downsample rewrite for proper dilation in 3x3 and avg_pool cases
5 years ago
xception.py
Big move, layer modules and fn to timm/models/layers
5 years ago