pytorch-image-models/timm/models
Ross Wightman f81cbdcca9
Merge pull request #2274 from huggingface/bulk_runner_tweaks
Better all res resolution for bulk runner
2024-09-03 12:11:56 -07:00
..
_pruned
layers
__init__.py Better all res resolution for bulk runner 2024-08-26 22:28:01 -07:00
_builder.py exception for unknown arg should have been TypeError 2024-08-12 16:09:36 -07:00
_efficientnet_blocks.py
_efficientnet_builder.py
_factory.py
_features.py
_features_fx.py
_helpers.py exception for unknown arg should have been TypeError 2024-08-12 16:09:36 -07:00
_hub.py exception for unknown arg should have been TypeError 2024-08-12 16:09:36 -07:00
_manipulate.py
_pretrained.py
_prune.py
_registry.py Better all res resolution for bulk runner 2024-08-26 22:28:01 -07:00
beit.py
byoanet.py
byobnet.py
cait.py
coat.py
convit.py
convmixer.py
convnext.py
crossvit.py
cspnet.py
davit.py
deit.py
densenet.py
dla.py
dpn.py
edgenext.py
efficientformer.py
efficientformer_v2.py Add a few missing __all__ entries. 2024-08-07 09:35:51 -07:00
efficientnet.py mobilenet_edgetpu can use group_size override, more consistency in arg wrap/sadface w/ extra group_size arg 2024-08-22 11:44:02 -07:00
efficientvit_mit.py
efficientvit_msra.py
eva.py
factory.py
fastvit.py Add a few missing __all__ entries. 2024-08-07 09:35:51 -07:00
features.py
focalnet.py
fx_features.py
gcvit.py
ghostnet.py
hardcorenas.py
helpers.py
hgnet.py Added gradient checkpointing to hgnet 2024-08-07 09:22:51 -07:00
hiera.py Remaining Hiera sbb weights uploaded 2024-08-21 10:06:27 -07:00
hieradet_sam2.py Add layer scale to hieradet 2024-08-21 11:23:39 -07:00
hrnet.py
hub.py
inception_next.py
inception_resnet_v2.py
inception_v3.py
inception_v4.py
levit.py
maxxvit.py
metaformer.py
mlp_mixer.py
mobilenetv3.py Add MobileNetV3 RA4 (mnv4 recipe) weights 2024-09-02 13:10:34 -07:00
mobilevit.py
mvitv2.py
nasnet.py
nest.py
nextvit.py Add a few missing __all__ entries. 2024-08-07 09:35:51 -07:00
nfnet.py Fix consistency, testing for forward_head w/ pre_logits, reset_classifier, models with pre_logits size != unpooled feature size 2024-06-07 13:53:00 -07:00
pit.py
pnasnet.py
pvt_v2.py
rdnet.py Remove eval to avoid safety concerns 2024-08-07 14:54:13 -07:00
registry.py
regnet.py
repghost.py
repvit.py
res2net.py
resnest.py
resnet.py Add resnet50d and efficientnet_b1 ra4 (mnv4) hparam weights 2024-08-16 12:10:00 -07:00
resnetv2.py
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 Fix #2272 2024-09-02 13:20:05 -07:00
vision_transformer_hybrid.py
vision_transformer_relpos.py
vision_transformer_sam.py proj_drop wasn't being called in sam vit 2024-08-02 16:12:02 -07:00
vitamin.py
volo.py
vovnet.py
xception.py
xception_aligned.py
xcit.py