pytorch-image-models/timm/models
Ross Wightman 150356c493 Fix unfortunate selecsls case bug caused by aggressive IDE rename 2023-08-03 10:37:06 -07:00
..
_pruned
layers Update README, prep for version 0.9.0 release 2023-05-11 15:22:50 -07:00
__init__.py Add RepViT models 2023-07-21 14:56:53 +08:00
_builder.py Throw when pretrained weights not available and pretrained=True (principle of least surprise). 2023-05-10 10:44:34 -07:00
_efficientnet_blocks.py Clean more feature extract issues 2023-06-14 14:46:22 -07:00
_efficientnet_builder.py Clean more feature extract issues 2023-06-14 14:46:22 -07:00
_factory.py Merge pull request #1643 from nateraw/docstrings-update 2023-05-09 21:33:20 -07:00
_features.py Clean more feature extract issues 2023-06-14 14:46:22 -07:00
_features_fx.py Clean more feature extract issues 2023-06-14 14:46:22 -07:00
_helpers.py
_hub.py More models w/ multi-weight support, moving to HF hub. Removing inplace_abn from all models including TResNet 2023-04-20 22:44:49 -07:00
_manipulate.py
_pretrained.py Allow passing state_dict directly via pretrained cfg mechanism as an override 2023-05-08 15:15:44 -07:00
_prune.py
_registry.py
beit.py Comment out beit url, no longer valid as now require long query string, leave for reference, must use HF hub now. 2023-08-03 10:00:46 -07:00
byoanet.py Missed hub_id entries for byoanet models 2023-05-10 15:58:55 -07:00
byobnet.py Add typing to all model entrypoint fns, add old cache check env var to builder 2023-05-08 08:52:38 -07:00
cait.py Add typing to all model entrypoint fns, add old cache check env var to builder 2023-05-08 08:52:38 -07:00
coat.py Add typing to all model entrypoint fns, add old cache check env var to builder 2023-05-08 08:52:38 -07:00
convit.py Add typing to all model entrypoint fns, add old cache check env var to builder 2023-05-08 08:52:38 -07:00
convmixer.py Add typing to all model entrypoint fns, add old cache check env var to builder 2023-05-08 08:52:38 -07:00
convnext.py Add typing to all model entrypoint fns, add old cache check env var to builder 2023-05-08 08:52:38 -07:00
crossvit.py Add typing to all model entrypoint fns, add old cache check env var to builder 2023-05-08 08:52:38 -07:00
cspnet.py Add typing to all model entrypoint fns, add old cache check env var to builder 2023-05-08 08:52:38 -07:00
davit.py Add typing to all model entrypoint fns, add old cache check env var to builder 2023-05-08 08:52:38 -07:00
deit.py Tweak DinoV2 add, add MAE ViT weights, add initial intermediate layer getter experiment 2023-05-09 17:59:22 -07:00
densenet.py Add typing to all model entrypoint fns, add old cache check env var to builder 2023-05-08 08:52:38 -07:00
dla.py Missed a few pretrained tags... 2023-05-10 12:16:30 -07:00
dpn.py Add typing to all model entrypoint fns, add old cache check env var to builder 2023-05-08 08:52:38 -07:00
edgenext.py Add typing to all model entrypoint fns, add old cache check env var to builder 2023-05-08 08:52:38 -07:00
efficientformer.py Add typing to all model entrypoint fns, add old cache check env var to builder 2023-05-08 08:52:38 -07:00
efficientformer_v2.py Add typing to all model entrypoint fns, add old cache check env var to builder 2023-05-08 08:52:38 -07:00
efficientnet.py Clean more feature extract issues 2023-06-14 14:46:22 -07:00
eva.py use torch.jit.Final instead of Final for beit, eva 2023-05-12 09:12:14 -07:00
factory.py
features.py
focalnet.py Add typing to all model entrypoint fns, add old cache check env var to builder 2023-05-08 08:52:38 -07:00
fx_features.py
gcvit.py Missed a few pretrained tags... 2023-05-10 12:16:30 -07:00
ghostnet.py Add typing to all model entrypoint fns, add old cache check env var to builder 2023-05-08 08:52:38 -07:00
hardcorenas.py Add typing to all model entrypoint fns, add old cache check env var to builder 2023-05-08 08:52:38 -07:00
helpers.py
hrnet.py Clean more feature extract issues 2023-06-14 14:46:22 -07:00
hub.py Update hub.py 2023-05-12 21:48:46 +02:00
inception_resnet_v2.py Remove label offsets and remove old weight url for 1001 class (background + in1k) TF origin weights 2023-05-09 18:00:41 -07:00
inception_v3.py Remove label offsets and remove old weight url for 1001 class (background + in1k) TF origin weights 2023-05-09 18:00:41 -07:00
inception_v4.py Remove label offsets and remove old weight url for 1001 class (background + in1k) TF origin weights 2023-05-09 18:00:41 -07:00
levit.py Add typing to all model entrypoint fns, add old cache check env var to builder 2023-05-08 08:52:38 -07:00
maxxvit.py Add 21k maxvit tf weights 2023-05-10 18:23:32 -07:00
metaformer.py Add typing to all model entrypoint fns, add old cache check env var to builder 2023-05-08 08:52:38 -07:00
mlp_mixer.py Add typing to all model entrypoint fns, add old cache check env var to builder 2023-05-08 08:52:38 -07:00
mobilenetv3.py Clean more feature extract issues 2023-06-14 14:46:22 -07:00
mobilevit.py Add typing to all model entrypoint fns, add old cache check env var to builder 2023-05-08 08:52:38 -07:00
mvitv2.py Add typing to all model entrypoint fns, add old cache check env var to builder 2023-05-08 08:52:38 -07:00
nasnet.py Remove label offsets and remove old weight url for 1001 class (background + in1k) TF origin weights 2023-05-09 18:00:41 -07:00
nest.py Add typing to all model entrypoint fns, add old cache check env var to builder 2023-05-08 08:52:38 -07:00
nfnet.py Add typing to all model entrypoint fns, add old cache check env var to builder 2023-05-08 08:52:38 -07:00
pit.py Add typing to all model entrypoint fns, add old cache check env var to builder 2023-05-08 08:52:38 -07:00
pnasnet.py Remove label offsets and remove old weight url for 1001 class (background + in1k) TF origin weights 2023-05-09 18:00:41 -07:00
pvt_v2.py Missed a few pretrained tags... 2023-05-10 12:16:30 -07:00
registry.py
regnet.py Add typing to all model entrypoint fns, add old cache check env var to builder 2023-05-08 08:52:38 -07:00
repvit.py Two small fixes, num_classes in base class, add model tag 2023-07-26 13:18:49 -07:00
res2net.py Add typing to all model entrypoint fns, add old cache check env var to builder 2023-05-08 08:52:38 -07:00
resnest.py Add typing to all model entrypoint fns, add old cache check env var to builder 2023-05-08 08:52:38 -07:00
resnet.py Add proper pool size for new resnexts 2023-07-26 14:36:03 -07:00
resnetv2.py Add typing to all model entrypoint fns, add old cache check env var to builder 2023-05-08 08:52:38 -07:00
rexnet.py Add typing to all model entrypoint fns, add old cache check env var to builder 2023-05-08 08:52:38 -07:00
selecsls.py Fix unfortunate selecsls case bug caused by aggressive IDE rename 2023-08-03 10:37:06 -07:00
senet.py Add typing to all model entrypoint fns, add old cache check env var to builder 2023-05-08 08:52:38 -07:00
sequencer.py Add typing to all model entrypoint fns, add old cache check env var to builder 2023-05-08 08:52:38 -07:00
sknet.py Add typing to all model entrypoint fns, add old cache check env var to builder 2023-05-08 08:52:38 -07:00
swin_transformer.py Add typing to all model entrypoint fns, add old cache check env var to builder 2023-05-08 08:52:38 -07:00
swin_transformer_v2.py Add typing to all model entrypoint fns, add old cache check env var to builder 2023-05-08 08:52:38 -07:00
swin_transformer_v2_cr.py Add typing to all model entrypoint fns, add old cache check env var to builder 2023-05-08 08:52:38 -07:00
tnt.py Add typing to all model entrypoint fns, add old cache check env var to builder 2023-05-08 08:52:38 -07:00
tresnet.py Add typing to all model entrypoint fns, add old cache check env var to builder 2023-05-08 08:52:38 -07:00
twins.py Add typing to all model entrypoint fns, add old cache check env var to builder 2023-05-08 08:52:38 -07:00
vgg.py Final push to get remaining models using multi-weight pretrained configs, almost all weights on HF hub 2023-04-26 15:52:13 -07:00
visformer.py F.sdpa for visformer fails w/o contiguous on qkv, make experimental 2023-05-11 11:37:37 -07:00
vision_transformer.py Fix ijepa vit issue with 448 model, minor formatting fixes 2023-07-26 20:46:27 -07:00
vision_transformer_hybrid.py Tweak DinoV2 add, add MAE ViT weights, add initial intermediate layer getter experiment 2023-05-09 17:59:22 -07:00
vision_transformer_relpos.py Add typing to all model entrypoint fns, add old cache check env var to builder 2023-05-08 08:52:38 -07:00
vision_transformer_sam.py ‘add ViT weight from I-JEPA pretrain’ 2023-06-14 22:30:31 +08:00
volo.py Replace deprecated NumPy aliases of builtin types 2023-07-03 22:24:25 +03:00
vovnet.py Add typing to all model entrypoint fns, add old cache check env var to builder 2023-05-08 08:52:38 -07:00
xception.py Add typing to all model entrypoint fns, add old cache check env var to builder 2023-05-08 08:52:38 -07:00
xception_aligned.py Add typing to all model entrypoint fns, add old cache check env var to builder 2023-05-08 08:52:38 -07:00
xcit.py Add typing to all model entrypoint fns, add old cache check env var to builder 2023-05-08 08:52:38 -07:00