Ross Wightman f45de37690 Merge branch 'master' into levit_visformer_rednet		2021-05-22 16:34:31 -07:00
..
layers	Merge branch 'master' into levit_visformer_rednet	2021-05-22 16:34:31 -07:00
pruned	…
__init__.py	Merge branch 'master' into levit_visformer_rednet	2021-05-22 16:34:31 -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	Fix crop_pct for cait models.	2021-05-11 10:50:15 -07:00
coat.py	Rethink name of patch embed grid info	2021-05-06 14:08:20 -07:00
convit.py	ConViT cleanup, fix torchscript, bit of reformatting, reuse existing layers.	2021-05-21 17:04:23 -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	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
levit.py	Add levit, levit_c, and visformer model defs. Largely untested and not finished cleanup.	2021-05-14 17:16:34 -07:00
levitc.py	Add levit, levit_c, and visformer model defs. Largely untested and not finished cleanup.	2021-05-14 17:16:34 -07:00
mlp_mixer.py	mixer_b16_224_miil, mixer_b16_224_miil_in21k models	2021-05-20 10:35:50 +03: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 prelim efficientnet_v2s weights from 224x224 train, eval 83.3 @ 288. Add eca_nfnet_l1 weights, train at 256, eval 84 @ 320.	2021-04-11 23:26:13 -07:00
pit.py	Add Swin Transformer models from https://github.com/microsoft/Swin-Transformer	2021-04-13 12:17:21 -07:00
pnasnet.py	Support for huggingface hub via create_model and default_cfgs.	2021-03-16 22:48:26 -07:00
registry.py	Add default_cfg options for min_input_size / fixed_input_size, queries in model registry, and use for testing self-attn models	2021-04-12 11:54:22 -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	extend positional embedding resizing functionality to tnt	2021-05-20 11:55:48 +01: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	Update twin weights to a copy in GitHub releases for faster dl. Tweak model class comment.	2021-05-21 16:23:14 -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 temporary default_cfgs to visformer models so they pass tests	2021-05-15 08:41:31 -07:00
vision_transformer.py	Minor adjustment, mutable default arg, extra check of valid len...	2021-05-21 17:20: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