pytorch-image-models/timm/models
Ross Wightman 2b49ab7a36 Fix ResNetV2 pretrained classifier issue. Fixes #540 2021-04-03 11:18:12 -07:00
..
layers Merge remote-tracking branch 'origin/benchmark-fixes-vit_hybrids' into pit_and_vit_update 2021-04-01 12:15:34 -07:00
pruned
__init__.py Merge remote-tracking branch 'origin/benchmark-fixes-vit_hybrids' into pit_and_vit_update 2021-04-01 12:15:34 -07:00
byobnet.py Improve RegVGG block identity/vs non for clariy and fix attn usage. Add comments. 2021-02-10 14:40:29 -08:00
cspnet.py Support for huggingface hub via create_model and default_cfgs. 2021-03-16 22:48:26 -07:00
densenet.py Support for huggingface hub via create_model and default_cfgs. 2021-03-16 22:48:26 -07:00
dla.py Support for huggingface hub via create_model and default_cfgs. 2021-03-16 22:48:26 -07:00
dpn.py Support for huggingface hub via create_model and default_cfgs. 2021-03-16 22:48:26 -07:00
efficientnet.py EfficientNet-V2S preliminary model def (for experimentation) 2021-04-02 09:36:51 -07:00
efficientnet_blocks.py Move stride in EdgeResidual block to 3x3 expansion conv. Fix #414 2021-02-07 22:10:18 -08:00
efficientnet_builder.py
factory.py A few minor fixes and bit more cleanup on the huggingface hub integration. 2021-03-17 13:18:52 -07:00
features.py
gluon_resnet.py Support for huggingface hub via create_model and default_cfgs. 2021-03-16 22:48:26 -07:00
gluon_xception.py Support for huggingface hub via create_model and default_cfgs. 2021-03-16 22:48:26 -07:00
hardcorenas.py Support for huggingface hub via create_model and default_cfgs. 2021-03-16 22:48:26 -07:00
helpers.py Fix pos_embed scaling for ViT and num_classes != 1000 for pretrained distilled deit and pit models. Fix #426 and fix #433 2021-03-31 23:11:28 -07:00
hrnet.py Support for huggingface hub via create_model and default_cfgs. 2021-03-16 22:48:26 -07:00
hub.py Change huggingface hub revision delimiter to '@', add hf_hub reference for eca_nfnet_l0 model as an example. 2021-03-18 11:38:06 -07:00
inception_resnet_v2.py Support for huggingface hub via create_model and default_cfgs. 2021-03-16 22:48:26 -07:00
inception_v3.py Support for huggingface hub via create_model and default_cfgs. 2021-03-16 22:48:26 -07:00
inception_v4.py Support for huggingface hub via create_model and default_cfgs. 2021-03-16 22:48:26 -07:00
mobilenetv3.py Support for huggingface hub via create_model and default_cfgs. 2021-03-16 22:48:26 -07:00
nasnet.py Support for huggingface hub via create_model and default_cfgs. 2021-03-16 22:48:26 -07:00
nfnet.py Wrong default_cfg pool_size for L1 2021-04-01 20:00:41 -07:00
pit.py Fix pos_embed scaling for ViT and num_classes != 1000 for pretrained distilled deit and pit models. Fix #426 and fix #433 2021-03-31 23:11:28 -07:00
pnasnet.py Support for huggingface hub via create_model and default_cfgs. 2021-03-16 22:48:26 -07:00
registry.py
regnet.py Add regnety_160 weights from DeiT teacher model, update that and my regnety_032 weights to use higher test size. 2021-04-01 16:41:04 -07:00
res2net.py Support for huggingface hub via create_model and default_cfgs. 2021-03-16 22:48:26 -07:00
resnest.py Support for huggingface hub via create_model and default_cfgs. 2021-03-16 22:48:26 -07:00
resnet.py Support for huggingface hub via create_model and default_cfgs. 2021-03-16 22:48:26 -07:00
resnetv2.py Fix ResNetV2 pretrained classifier issue. Fixes #540 2021-04-03 11:18:12 -07:00
rexnet.py Support for huggingface hub via create_model and default_cfgs. 2021-03-16 22:48:26 -07:00
selecsls.py Support for huggingface hub via create_model and default_cfgs. 2021-03-16 22:48:26 -07:00
senet.py Support for huggingface hub via create_model and default_cfgs. 2021-03-16 22:48:26 -07:00
sknet.py Support for huggingface hub via create_model and default_cfgs. 2021-03-16 22:48:26 -07:00
tnt.py Update TNT 2021-03-29 08:23:34 +08:00
tresnet.py Support for huggingface hub via create_model and default_cfgs. 2021-03-16 22:48:26 -07:00
vgg.py Support for huggingface hub via create_model and default_cfgs. 2021-03-16 22:48:26 -07:00
vision_transformer.py Merge jax and original weight init 2021-04-01 18:11:51 -07:00
vision_transformer_hybrid.py Update benchmark script to add precision arg. Fix some downstream (DeiT) compat issues with latest changes. Bump version to 0.4.7 2021-04-01 16:40:12 -07:00
vovnet.py Support for huggingface hub via create_model and default_cfgs. 2021-03-16 22:48:26 -07:00
xception.py Support for huggingface hub via create_model and default_cfgs. 2021-03-16 22:48:26 -07:00
xception_aligned.py Support for huggingface hub via create_model and default_cfgs. 2021-03-16 22:48:26 -07:00