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.
pytorch-image-models/timm/models
Ross Wightman a69c0e04f0
Fix pool size in cspnet
4 years ago
..
layers Add ReXNet w/ remapped weights, feature support 4 years ago
pruned added efficientnet pruned weights 4 years ago
__init__.py Rename csp.py -> cspnet.py 4 years ago
cspnet.py Fix pool size in cspnet 4 years ago
densenet.py Lots of changes to model creation helpers, close to finalizing feature extraction / interfaces 4 years ago
dla.py More models supporting feature extraction, xception, gluon_xception, inception_v3, inception_v4, pnasnet, nasnet, dla. Fix DLA unused projection params. 4 years ago
dpn.py Lots of changes to model creation helpers, close to finalizing feature extraction / interfaces 4 years ago
efficientnet.py Cleanup FeatureInfo getters, add TF models sourced Xception41/65/71 weights 4 years ago
efficientnet_blocks.py Cleanup EfficientNet/MobileNetV3 feature extraction a bit, only two tap locations now, small mobilenetv3 models work 4 years ago
efficientnet_builder.py Cleanup EfficientNet/MobileNetV3 feature extraction a bit, only two tap locations now, small mobilenetv3 models work 4 years ago
factory.py Add set_layer_config contextmgr to adjust all layer configs at once, use in create_module with new args. Remove a few old warning causing constant annotations for jit. 4 years ago
features.py Cleanup FeatureInfo getters, add TF models sourced Xception41/65/71 weights 4 years ago
gluon_resnet.py Lots of changes to model creation helpers, close to finalizing feature extraction / interfaces 4 years ago
gluon_xception.py Cleanup, refactoring of Feature extraction code, add tests, fix tests, non hook feature extraction working with torchscript 4 years ago
helpers.py Cleanup, refactoring of Feature extraction code, add tests, fix tests, non hook feature extraction working with torchscript 4 years ago
hrnet.py Add HRNet feature extraction, fix senet type, lower feature testing res to 96x96 4 years ago
inception_resnet_v2.py Lots of changes to model creation helpers, close to finalizing feature extraction / interfaces 4 years ago
inception_v3.py More models supporting feature extraction, xception, gluon_xception, inception_v3, inception_v4, pnasnet, nasnet, dla. Fix DLA unused projection params. 4 years ago
inception_v4.py More models supporting feature extraction, xception, gluon_xception, inception_v3, inception_v4, pnasnet, nasnet, dla. Fix DLA unused projection params. 4 years ago
mobilenetv3.py Cleanup FeatureInfo getters, add TF models sourced Xception41/65/71 weights 4 years ago
nasnet.py Cleanup, refactoring of Feature extraction code, add tests, fix tests, non hook feature extraction working with torchscript 4 years ago
pnasnet.py Cleanup, refactoring of Feature extraction code, add tests, fix tests, non hook feature extraction working with torchscript 4 years ago
registry.py Exclude EfficientNet-L2 models from test 4 years ago
regnet.py Lots of changes to model creation helpers, close to finalizing feature extraction / interfaces 4 years ago
res2net.py Lots of changes to model creation helpers, close to finalizing feature extraction / interfaces 4 years ago
resnest.py Merge branch 'master' into features 4 years ago
resnet.py Lots of changes to model creation helpers, close to finalizing feature extraction / interfaces 4 years ago
rexnet.py Add ReXNet w/ remapped weights, feature support 4 years ago
selecsls.py Cleanup, refactoring of Feature extraction code, add tests, fix tests, non hook feature extraction working with torchscript 4 years ago
senet.py Add HRNet feature extraction, fix senet type, lower feature testing res to 96x96 4 years ago
sknet.py Lots of changes to model creation helpers, close to finalizing feature extraction / interfaces 4 years ago
tresnet.py Cleanup, refactoring of Feature extraction code, add tests, fix tests, non hook feature extraction working with torchscript 4 years ago
vovnet.py Lots of changes to model creation helpers, close to finalizing feature extraction / interfaces 4 years ago
xception.py Cleanup, refactoring of Feature extraction code, add tests, fix tests, non hook feature extraction working with torchscript 4 years ago
xception_aligned.py Cleanup FeatureInfo getters, add TF models sourced Xception41/65/71 weights 4 years ago