mirrors
/
pytorch-image-models
mirror of https://github.com/huggingface/pytorch-image-models.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
pytorch-image-models/timm/models
History
Ross Wightman 855d6cc217 More dataset work including factories and a tensorflow datasets (TFDS) wrapper 
* Add parser/dataset factory methods for more flexible dataset & parser creation
* Add dataset parser that wraps TFDS image classification datasets
* Tweak num_classes handling bug for 21k models
* Add initial deit models so they can be benchmarked in next csv results runs
 		2021-01-15 17:26:20 -08:00
..
layers ResNetV2 pre-act and non-preact model, w/ BiT pretrained weights and support for ViT R50 model. Tweaks for in21k num_classes passing. More to do... tests failing. 2020-12-28 16:59:15 -08:00
pruned
…
__init__.py ResNetV2 pre-act and non-preact model, w/ BiT pretrained weights and support for ViT R50 model. Tweaks for in21k num_classes passing. More to do... tests failing. 2020-12-28 16:59:15 -08:00
cspnet.py Add CSPResNet50 weights, 79.6 top-1 at 256x256 2020-08-12 11:20:11 -07:00
densenet.py More uniform treatment of classifiers across all models, reduce code duplication. 2020-08-03 22:18:24 -07:00
dla.py More uniform treatment of classifiers across all models, reduce code duplication. 2020-08-03 22:18:24 -07:00
dpn.py Remove some old __future__ imports 2021-01-03 11:24:16 -08:00
efficientnet.py Bug in last mod for features_only default_cfg 2020-12-03 12:33:01 -08:00
efficientnet_blocks.py AdaptiveAvgPool2d -> mean((2,3)) for all SE/attn layers to avoid NaN with AMP + channels_last layout. See https://github.com/pytorch/pytorch/issues/43992 2020-09-01 16:05:32 -07:00
efficientnet_builder.py Merge branch 'logger' into features. Change 'logger' to '_logger'. 2020-07-27 18:00:46 -07:00
factory.py ResNetV2 pre-act and non-preact model, w/ BiT pretrained weights and support for ViT R50 model. Tweaks for in21k num_classes passing. More to do... tests failing. 2020-12-28 16:59:15 -08:00
features.py Fix some attributions, add copyrights to some file docstrings 2020-07-27 13:44:56 -07:00
gluon_resnet.py fix test_model_default_cfgs 2020-08-11 23:23:57 +09:00
gluon_xception.py More uniform treatment of classifiers across all models, reduce code duplication. 2020-08-03 22:18:24 -07:00
helpers.py More dataset work including factories and a tensorflow datasets (TFDS) wrapper 2021-01-15 17:26:20 -08:00
hrnet.py Bug in last mod for features_only default_cfg 2020-12-03 12:33:01 -08:00
inception_resnet_v2.py More uniform treatment of classifiers across all models, reduce code duplication. 2020-08-03 22:18:24 -07:00
inception_v3.py A few more changes for 0.3.2 maint release. Linear layer change for mobilenetv3 and inception_v3, support no bias for linear wrapper. 2020-11-30 16:19:52 -08:00
inception_v4.py More uniform treatment of classifiers across all models, reduce code duplication. 2020-08-03 22:18:24 -07:00
mobilenetv3.py Bug in last mod for features_only default_cfg 2020-12-03 12:33:01 -08:00
nasnet.py ResNetV2 pre-act and non-preact model, w/ BiT pretrained weights and support for ViT R50 model. Tweaks for in21k num_classes passing. More to do... tests failing. 2020-12-28 16:59:15 -08:00
pnasnet.py More uniform treatment of classifiers across all models, reduce code duplication. 2020-08-03 22:18:24 -07:00
registry.py Fix some attributions, add copyrights to some file docstrings 2020-07-27 13:44:56 -07:00
regnet.py Add DropPath (stochastic depth) to ReXNet and VoVNet. RegNet DropPath impl tweak and dedupe se args. 2020-09-24 18:20:36 -07:00
res2net.py Fix some checkpoint / model str regressions 2020-07-29 19:43:01 -07:00
resnest.py fix test_model_default_cfgs 2020-08-11 23:23:57 +09:00
resnet.py Add 320x320 model default cfgs for 101D and 152D ResNets. Add SEResNet-152D weights and 320x320 cfg. 2021-01-03 12:10:25 -08:00
resnetv2.py More dataset work including factories and a tensorflow datasets (TFDS) wrapper 2021-01-15 17:26:20 -08:00
rexnet.py Initial Vision Transformer impl w/ patch and hybrid variants. Refactor tuple helpers. 2020-10-13 13:33:44 -07:00
selecsls.py fix test_model_default_cfgs 2020-08-11 23:23:57 +09:00
senet.py AdaptiveAvgPool2d -> mean((2,3)) for all SE/attn layers to avoid NaN with AMP + channels_last layout. See https://github.com/pytorch/pytorch/issues/43992 2020-09-01 16:05:32 -07:00
sknet.py fix test_model_default_cfgs 2020-08-11 23:23:57 +09:00
tresnet.py Add 'fast' global pool option, remove redundant SEModule from tresnet, normal one is now 'fast' 2020-09-02 09:11:48 -07:00
vision_transformer.py More dataset work including factories and a tensorflow datasets (TFDS) wrapper 2021-01-15 17:26:20 -08:00
vovnet.py Add DropPath (stochastic depth) to ReXNet and VoVNet. RegNet DropPath impl tweak and dedupe se args. 2020-09-24 18:20:36 -07:00
xception.py More uniform treatment of classifiers across all models, reduce code duplication. 2020-08-03 22:18:24 -07:00
xception_aligned.py Initial Vision Transformer impl w/ patch and hybrid variants. Refactor tuple helpers. 2020-10-13 13:33:44 -07:00