Ross Wightman 4fbc32d3d0 Fix crop_pct for cait models.		2021-05-11 10:50:15 -07:00
..
layers	Rethink name of patch embed grid info	2021-05-06 14:08:20 -07:00
pruned	…
__init__.py	Merge branch 'master' into cait	2021-05-05 17:29:38 -07:00
byoanet.py	More adjustments to ByoaNet models for further experiments.	2021-05-06 13:59:44 -07:00
byobnet.py	Update ByoaNet comments. Fix first Steam feat chs for ByobNet.	2021-04-12 12:11:35 -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
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 updated EfficientNet-V2S weights, 83.8 @ 384x384 test. Add PyTorch trained EfficientNet-B4 weights, 83.4 @ 384x384 test. Tweak non TF EfficientNet B1-B4 train/test res scaling.	2021-04-19 10:42:56 -07:00
efficientnet_blocks.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
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	…
ghostnet.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
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
mlp_mixer.py	Move Mlp and PatchEmbed modules into layers. Being used in lots of models now...	2021-05-06 14:03:23 -07:00
mobilenetv3.py	A few miil weights naming tweaks to improve compat with model registry and filtering wildcards.	2021-04-28 10:33:02 -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	Move Mlp and PatchEmbed modules into layers. Being used in lots of models now...	2021-05-06 14:03:23 -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
vgg.py	Support for huggingface hub via create_model and default_cfgs.	2021-03-16 22:48:26 -07:00
vision_transformer.py	Move Mlp and PatchEmbed modules into layers. Being used in lots of models now...	2021-05-06 14:03:23 -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