pytorch-image-models/timm/models
Ross Wightman ac18adb9c3 Remove debug print from RexNet 2020-07-29 11:15:19 -07:00
..
layers Merge branch 'logger' into features. Change 'logger' to '_logger'. 2020-07-27 18:00:46 -07:00
pruned added efficientnet pruned weights 2020-05-03 09:29:45 +03:00
__init__.py Rename csp.py -> cspnet.py 2020-07-27 11:15:07 -07:00
cspnet.py Fix pool size in cspnet 2020-07-27 13:44:02 -07:00
densenet.py Lots of changes to model creation helpers, close to finalizing feature extraction / interfaces 2020-07-17 17:54:26 -07:00
dla.py More models supporting feature extraction, xception, gluon_xception, inception_v3, inception_v4, pnasnet, nasnet, dla. Fix DLA unused projection params. 2020-07-19 14:02:02 -07:00
dpn.py Fix some attributions, add copyrights to some file docstrings 2020-07-27 13:44:56 -07:00
efficientnet.py Add EfficientNet-Lite0 weights trained with this code by @hal-314, 75.484 top-1 2020-07-29 00:32:08 -07:00
efficientnet_blocks.py Fix some attributions, add copyrights to some file docstrings 2020-07-27 13:44:56 -07:00
efficientnet_builder.py Merge branch 'logger' into features. Change 'logger' to '_logger'. 2020-07-27 18:00:46 -07:00
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. 2020-06-02 21:06:10 -07:00
features.py Fix some attributions, add copyrights to some file docstrings 2020-07-27 13:44:56 -07:00
gluon_resnet.py Lots of changes to model creation helpers, close to finalizing feature extraction / interfaces 2020-07-17 17:54:26 -07:00
gluon_xception.py Fix some attributions, add copyrights to some file docstrings 2020-07-27 13:44:56 -07:00
helpers.py Merge branch 'logger' into features. Change 'logger' to '_logger'. 2020-07-27 18:00:46 -07:00
hrnet.py Merge branch 'logger' into features. Change 'logger' to '_logger'. 2020-07-27 18:00:46 -07:00
inception_resnet_v2.py Lots of changes to model creation helpers, close to finalizing feature extraction / interfaces 2020-07-17 17:54:26 -07:00
inception_v3.py Fix some attributions, add copyrights to some file docstrings 2020-07-27 13:44:56 -07:00
inception_v4.py More models supporting feature extraction, xception, gluon_xception, inception_v3, inception_v4, pnasnet, nasnet, dla. Fix DLA unused projection params. 2020-07-19 14:02:02 -07:00
mobilenetv3.py Fix some attributions, add copyrights to some file docstrings 2020-07-27 13:44:56 -07:00
nasnet.py Cleanup, refactoring of Feature extraction code, add tests, fix tests, non hook feature extraction working with torchscript 2020-07-20 16:10:31 -07:00
pnasnet.py Cleanup, refactoring of Feature extraction code, add tests, fix tests, non hook feature extraction working with torchscript 2020-07-20 16:10:31 -07:00
registry.py Fix some attributions, add copyrights to some file docstrings 2020-07-27 13:44:56 -07:00
regnet.py Fix some attributions, add copyrights to some file docstrings 2020-07-27 13:44:56 -07:00
res2net.py Lots of changes to model creation helpers, close to finalizing feature extraction / interfaces 2020-07-17 17:54:26 -07:00
resnest.py Merge branch 'master' into features 2020-07-20 16:11:38 -07:00
resnet.py Add new seresnet50 (non-legacy) model weights, 80.274 top-1 2020-07-29 00:17:42 -07:00
rexnet.py Remove debug print from RexNet 2020-07-29 11:15:19 -07:00
selecsls.py Cleanup, refactoring of Feature extraction code, add tests, fix tests, non hook feature extraction working with torchscript 2020-07-20 16:10:31 -07:00
senet.py Add HRNet feature extraction, fix senet type, lower feature testing res to 96x96 2020-07-21 17:39:29 -07:00
sknet.py Fix some attributions, add copyrights to some file docstrings 2020-07-27 13:44:56 -07:00
tresnet.py Cleanup, refactoring of Feature extraction code, add tests, fix tests, non hook feature extraction working with torchscript 2020-07-20 16:10:31 -07:00
vovnet.py Fix some attributions, add copyrights to some file docstrings 2020-07-27 13:44:56 -07:00
xception.py Cleanup, refactoring of Feature extraction code, add tests, fix tests, non hook feature extraction working with torchscript 2020-07-20 16:10:31 -07:00
xception_aligned.py Fix some attributions, add copyrights to some file docstrings 2020-07-27 13:44:56 -07:00