mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity
pytorch-image-models/timm/models
History
Ross Wightman f0e65e37b7 Fix NF-ResNet101 model defs
2021-01-30 23:26:19 -08:00
..
layers
Fix up a few details in NFResNet models, managed stable training. Add support for gamma gain to be applied in activation or ScaleStdConv. Some tweaks to ScaledStdConv.
2021-01-30 16:32:07 -08:00
pruned
added efficientnet pruned weights
2020-05-03 09:29:45 +03:00
__init__.py
Initial Normalizer-Free Reg/ResNet impl. A bit of related layer refactoring.
2021-01-27 22:06:57 -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_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
efficientnet.py
Bug in last mod for features_only default_cfg
2020-12-03 12:33:01 -08: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
Fix potential issue with change to num_classes arg in train/validate.py defaulting to None (rely on model def / default_cfg)
2021-01-25 11:53:34 -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
nfnet.py
Fix NF-ResNet101 model defs
2021-01-30 23:26:19 -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
Convert samples/targets in ParserImageInTar to numpy arrays, slightly less mem usage for massive datasets. Add a few more se/eca model defs to resnet.py
2021-01-22 16:54:33 -08:00
resnetv2.py
Initial Normalizer-Free Reg/ResNet impl. A bit of related layer refactoring.
2021-01-27 22:06:57 -08:00
rexnet.py
Initial Normalizer-Free Reg/ResNet impl. A bit of related layer refactoring.
2021-01-27 22:06:57 -08: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
Initial Normalizer-Free Reg/ResNet impl. A bit of related layer refactoring.
2021-01-27 22:06:57 -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_aligned.py
Initial Vision Transformer impl w/ patch and hybrid variants. Refactor tuple helpers.
2020-10-13 13:33:44 -07:00
xception.py
More uniform treatment of classifiers across all models, reduce code duplication.
2020-08-03 22:18:24 -07:00