pytorch-image-models/timm/models
Ross Wightman b0068ba5d0 Switch hf hub entries for new aimv2 / dfn weights to point to timm locations. Undo forced device for SDR linspace, part of another change. 2024-12-30 19:24:21 -08:00
..
_pruned
layers
__init__.py
_builder.py More tweaks to docstrings for hub/builder 2024-12-06 10:25:06 -08:00
_efficientnet_blocks.py
_efficientnet_builder.py
_factory.py Add cache_dir example 2024-12-06 10:39:13 -08:00
_features.py
_features_fx.py
_helpers.py Punch cache_dir through model factory / builder / pretrain helpers. Improve some annotations in related code. 2024-12-06 10:25:06 -08:00
_hub.py Add support for tag, license customization through push_to_hub 2024-12-27 14:00:41 -08:00
_manipulate.py
_pretrained.py
_prune.py
_registry.py
beit.py
byoanet.py
byobnet.py Add more pali(2) weights. Switch rest of models adapting open_clip weights to their own weight instances. 2024-12-27 14:00:41 -08:00
cait.py
coat.py
convit.py
convmixer.py
convnext.py Fixing RmsNorm to fix #2380 and noticed with aimv2 when comparing outputs. Still some work to do, need to look at AMP / fast mode behaviour, dispatch to torch when possible. Add SimpleNorm for 'LayerNorm w/o centering and bias' 2024-12-30 19:24:21 -08:00
crossvit.py
cspnet.py CS3-DarkNet Small (Focus) w/ RA4 recipe. Fix #2122 2024-11-22 16:31:45 -08:00
davit.py fix feature_info.reduction 2024-12-18 21:12:40 +08:00
deit.py
densenet.py
dla.py
dpn.py
edgenext.py
efficientformer.py fix feature_info.reduction 2024-12-18 21:12:40 +08:00
efficientformer_v2.py
efficientnet.py
efficientvit_mit.py
efficientvit_msra.py
eva.py Add more pali(2) weights. Switch rest of models adapting open_clip weights to their own weight instances. 2024-12-27 14:00:41 -08:00
factory.py
fastvit.py
features.py
focalnet.py
fx_features.py
gcvit.py
ghostnet.py
hardcorenas.py
helpers.py
hgnet.py
hiera.py
hieradet_sam2.py Add more pali(2) weights. Switch rest of models adapting open_clip weights to their own weight instances. 2024-12-27 14:00:41 -08:00
hrnet.py
hub.py
inception_next.py Rename inception_next_atto pretrained str 2024-12-06 10:36:47 -08:00
inception_resnet_v2.py
inception_v3.py
inception_v4.py
levit.py
mambaout.py
maxxvit.py
metaformer.py fix feature_info.reduction 2024-12-18 21:12:40 +08:00
mlp_mixer.py
mobilenetv3.py Add a different mnv4 conv-small weight 2024-12-02 16:14:37 -08:00
mobilevit.py
mvitv2.py
nasnet.py
nest.py
nextvit.py
nfnet.py
pit.py
pnasnet.py
pvt_v2.py
rdnet.py
registry.py
regnet.py
repghost.py
repvit.py
res2net.py
resnest.py
resnet.py Update torchvision resnet legacy weight urls in resnet.py 2024-11-26 15:53:54 +01:00
resnetv2.py Typo 2024-11-17 13:54:50 -08:00
rexnet.py
selecsls.py
senet.py
sequencer.py
sknet.py
swin_transformer.py
swin_transformer_v2.py
swin_transformer_v2_cr.py
tiny_vit.py
tnt.py
tresnet.py
twins.py
vgg.py
visformer.py
vision_transformer.py Switch hf hub entries for new aimv2 / dfn weights to point to timm locations. Undo forced device for SDR linspace, part of another change. 2024-12-30 19:24:21 -08:00
vision_transformer_hybrid.py
vision_transformer_relpos.py
vision_transformer_sam.py
vitamin.py Add bias arg to Vitamin GeGLU 2024-12-30 19:24:21 -08:00
volo.py
vovnet.py
xception.py
xception_aligned.py
xcit.py