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
5a0a8de7e3
ResNet updates:
...
* remove redundant GluonResNet model/blocks and use the code in ResNet for Gluon weights
* change SEModules back to using AdaptiveAvgPool instead of mean, PyTorch issue long fixed
5 years ago
..
__init__.py
Bring EfficientNet and MobileNetV3 up to date with my gen-efficientnet repo
5 years ago
activations.py
Bring EfficientNet and MobileNetV3 up to date with my gen-efficientnet repo
5 years ago
adaptive_avgmax_pool.py
Remove all prints, change most to logging calls, tweak alignment of batch logs, improve setup.py
5 years ago
conv2d_layers.py
Bring EfficientNet and MobileNetV3 up to date with my gen-efficientnet repo
5 years ago
densenet.py
Fix typo in Densenet default resolutions
5 years ago
dla.py
Add Res2Net and DLA models w/ pretrained weights. Update sotabench.
5 years ago
dpn.py
Register dpn107
5 years ago
efficientnet_blocks.py
Bring EfficientNet and MobileNetV3 up to date with my gen-efficientnet repo
5 years ago
efficientnet_builder.py
Bring EfficientNet and MobileNetV3 up to date with my gen-efficientnet repo
5 years ago
factory.py
Pass drop connect arg through to EfficientNet models
5 years ago
feature_hooks.py
Bring EfficientNet and MobileNetV3 up to date with my gen-efficientnet repo
5 years ago
gen_efficientnet.py
Bring EfficientNet and MobileNetV3 up to date with my gen-efficientnet repo
5 years ago
gluon_resnet.py
ResNet updates:
5 years ago
gluon_xception.py
A few more additions to Gluon Xception models to match interface of others.
5 years ago
helpers.py
Add CondConv support for EfficientNet into WIP for GenEfficientNet Feature extraction setup
5 years ago
hrnet.py
Initial HRNet classification model commit
5 years ago
inception_resnet_v2.py
Add model registry and model listing fns, refactor model_factory/create_model fn
5 years ago
inception_v3.py
Update version, results csv files, and move remaining dropbox weights to github
5 years ago
inception_v4.py
Add model registry and model listing fns, refactor model_factory/create_model fn
5 years ago
median_pool.py
Big re-org, working towards making pip/module as 'timm'
5 years ago
mobilenetv3.py
Bring EfficientNet and MobileNetV3 up to date with my gen-efficientnet repo
5 years ago
nasnet.py
Add NASNet-Large model
5 years ago
pnasnet.py
Add model registry and model listing fns, refactor model_factory/create_model fn
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
Fix Mobilenet V3 model name for sotabench. Minor res2net cleanup.
5 years ago
resnet.py
ResNet updates:
5 years ago
senet.py
ResNet updates:
5 years ago
test_time_pool.py
Remove all prints, change most to logging calls, tweak alignment of batch logs, improve setup.py
5 years ago
xception.py
Add model registry and model listing fns, refactor model_factory/create_model fn
5 years ago