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 11ae795e99 Redo LeViT attention bias caching in a way that works with both torchscript and DataParallel 2021-05-25 10:15:32 -07:00
..
layers Merge branch 'master' into levit_visformer_rednet 2021-05-22 16:34:31 -07:00
pruned
…
__init__.py Expand scope of testing for non-std vision transformer / mlp models. Some related cleanup and create fn cleanup for all vision transformer and mlp models. More CoaT weights. 2021-05-24 21:13:26 -07:00
byoanet.py Another set of byoanet models w/ ECA channel + SA + groups 2021-05-15 22:56:51 -07:00
byobnet.py Another set of byoanet models w/ ECA channel + SA + groups 2021-05-15 22:56:51 -07:00
cait.py Expand scope of testing for non-std vision transformer / mlp models. Some related cleanup and create fn cleanup for all vision transformer and mlp models. More CoaT weights. 2021-05-24 21:13:26 -07:00
coat.py Expand scope of testing for non-std vision transformer / mlp models. Some related cleanup and create fn cleanup for all vision transformer and mlp models. More CoaT weights. 2021-05-24 21:13:26 -07:00
convit.py Expand scope of testing for non-std vision transformer / mlp models. Some related cleanup and create fn cleanup for all vision transformer and mlp models. More CoaT weights. 2021-05-24 21:13:26 -07: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 Fix incorrect name of shortcut/identity paths in many residual nets. Inherited from naming in old old torchvision, long fixed there. 2021-05-04 21:27:15 -07:00
dpn.py Support for huggingface hub via create_model and default_cfgs. 2021-03-16 22:48:26 -07:00
efficientnet.py Add efficientnetv2_rw_m weights trained in PyTorch. 84.8 top-1 @ 416 test. 53M params. 2021-05-21 21:16:25 -07:00
efficientnet_blocks.py EfficientNetV2 official impl w/ weights ported from TF. Cleanup/refactor of related EfficientNet classes and models. 2021-05-14 15:50:00 -07:00
efficientnet_builder.py Prep for effcientnetv2_rw_m model weights that started training before official release.. 2021-05-21 17:47:49 -07:00
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 Fix some attributions, add copyrights to some file docstrings 2020-07-27 13:44:56 -07:00
ghostnet.py EfficientNetV2 official impl w/ weights ported from TF. Cleanup/refactor of related EfficientNet classes and models. 2021-05-14 15:50:00 -07:00
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 EfficientNetV2 official impl w/ weights ported from TF. Cleanup/refactor of related EfficientNet classes and models. 2021-05-14 15:50:00 -07:00
helpers.py Expand scope of testing for non-std vision transformer / mlp models. Some related cleanup and create fn cleanup for all vision transformer and mlp models. More CoaT weights. 2021-05-24 21:13:26 -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
levit.py Redo LeViT attention bias caching in a way that works with both torchscript and DataParallel 2021-05-25 10:15:32 -07:00
mlp_mixer.py Expand scope of testing for non-std vision transformer / mlp models. Some related cleanup and create fn cleanup for all vision transformer and mlp models. More CoaT weights. 2021-05-24 21:13:26 -07:00
mobilenetv3.py EfficientNetV2 official impl w/ weights ported from TF. Cleanup/refactor of related EfficientNet classes and models. 2021-05-14 15:50:00 -07:00
nasnet.py Support for huggingface hub via create_model and default_cfgs. 2021-03-16 22:48:26 -07:00
nfnet.py Add eca_nfnet_l2/l3 defs for future training 2021-05-22 21:55:37 -07:00
pit.py Expand scope of testing for non-std vision transformer / mlp models. Some related cleanup and create fn cleanup for all vision transformer and mlp models. More CoaT weights. 2021-05-24 21:13:26 -07:00
pnasnet.py Support for huggingface hub via create_model and default_cfgs. 2021-03-16 22:48:26 -07:00
registry.py Filter test models before creation for backward/torchscript tests 2021-05-25 10:14:45 -07:00
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 Fix incorrect name of shortcut/identity paths in many residual nets. Inherited from naming in old old torchvision, long fixed there. 2021-05-04 21:27:15 -07:00
resnest.py Fix incorrect name of shortcut/identity paths in many residual nets. Inherited from naming in old old torchvision, long fixed there. 2021-05-04 21:27:15 -07:00
resnet.py Move Mlp and PatchEmbed modules into layers. Being used in lots of models now... 2021-05-06 14:03:23 -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 Fix incorrect name of shortcut/identity paths in many residual nets. Inherited from naming in old old torchvision, long fixed there. 2021-05-04 21:27:15 -07:00
sknet.py Fix incorrect name of shortcut/identity paths in many residual nets. Inherited from naming in old old torchvision, long fixed there. 2021-05-04 21:27:15 -07:00
swin_transformer.py Rethink name of patch embed grid info 2021-05-06 14:08:20 -07:00
tnt.py Expand scope of testing for non-std vision transformer / mlp models. Some related cleanup and create fn cleanup for all vision transformer and mlp models. More CoaT weights. 2021-05-24 21:13:26 -07:00
tresnet.py Fix incorrect name of shortcut/identity paths in many residual nets. Inherited from naming in old old torchvision, long fixed there. 2021-05-04 21:27:15 -07:00
twins.py Expand scope of testing for non-std vision transformer / mlp models. Some related cleanup and create fn cleanup for all vision transformer and mlp models. More CoaT weights. 2021-05-24 21:13:26 -07:00
vgg.py Support for huggingface hub via create_model and default_cfgs. 2021-03-16 22:48:26 -07:00
visformer.py Add Visformer-small weighs, tweak torchscript jit test img size. 2021-05-24 22:50:12 -07:00
vision_transformer.py Expand scope of testing for non-std vision transformer / mlp models. Some related cleanup and create fn cleanup for all vision transformer and mlp models. More CoaT weights. 2021-05-24 21:13:26 -07:00
vision_transformer_hybrid.py Expand scope of testing for non-std vision transformer / mlp models. Some related cleanup and create fn cleanup for all vision transformer and mlp models. More CoaT weights. 2021-05-24 21:13:26 -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