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 110a7c4982 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
..
layers
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
pruned
…
__init__.py
Rename csp.py -> cspnet.py
2020-07-27 11:15:07 -07: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
More uniform treatment of classifiers across all models, reduce code duplication.
2020-08-03 22:18:24 -07: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
Add new EfficientNet-B3 and RegNetY-3.2GF weights, both just over 82 top-1
2020-08-11 14:18:51 -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
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 a few more issues related to #216 w/ TResNet (space2depth) and FP16 weights in wide resnets. Also don't completely dump pretrained weights in in_chans != 1 or 3 cases.
2020-08-11 18:57:47 -07:00
hrnet.py
More uniform treatment of classifiers across all models, reduce code duplication.
2020-08-03 22:18:24 -07: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
fix test_model_default_cfgs
2020-08-11 23:23:57 +09: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
Fix MobileNetV3 crash with global_pool='', output consistent with other models but not equivalent due to efficient head.
2020-08-18 14:11:30 -07:00
nasnet.py
More uniform treatment of classifiers across all models, reduce code duplication.
2020-08-03 22:18:24 -07: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 new EfficientNet-B3 and RegNetY-3.2GF weights, both just over 82 top-1
2020-08-11 14:18:51 -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
fix test_model_default_cfgs
2020-08-11 23:23:57 +09:00
rexnet.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
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
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
vovnet.py
Fix some attributions, add copyrights to some file docstrings
2020-07-27 13:44:56 -07:00
xception_aligned.py
fix test_model_default_cfgs
2020-08-11 23:23:57 +09:00
xception.py
More uniform treatment of classifiers across all models, reduce code duplication.
2020-08-03 22:18:24 -07:00