Ross Wightman bd0f79153b Add weights only flag to avoid warning, try to keep bwd compat. Default to True for remote load of pretrained weights, keep False for local checkpoing load to avoid training checkpoint breaks.. fix #2249

2024-08-12 11:41:41 -07:00

..

_pruned

…

layers

…

__init__.py

add rdnet

2024-08-01 14:49:21 +09:00

_builder.py

Add weights only flag to avoid warning, try to keep bwd compat. Default to True for remote load of pretrained weights, keep False for local checkpoing load to avoid training checkpoint breaks.. fix #2249

2024-08-12 11:41:41 -07:00

_efficientnet_blocks.py

…

_efficientnet_builder.py

Add xavier_uniform init of MNVC hybrid attention modules. Small improvement in training stability.

2024-07-26 17:03:40 -07:00

_factory.py

…

_features_fx.py

…

_features.py

…

_helpers.py

Add weights only flag to avoid warning, try to keep bwd compat. Default to True for remote load of pretrained weights, keep False for local checkpoing load to avoid training checkpoint breaks.. fix #2249

2024-08-12 11:41:41 -07:00

_hub.py

Add weights only flag to avoid warning, try to keep bwd compat. Default to True for remote load of pretrained weights, keep False for local checkpoing load to avoid training checkpoint breaks.. fix #2249

2024-08-12 11:41:41 -07:00

_manipulate.py

…

_pretrained.py

…

_prune.py

…

_registry.py

…

beit.py

Merge pull request #2225 from huggingface/small_things

2024-07-25 20:29:13 -07:00

byoanet.py

…

byobnet.py

Merge pull request #2225 from huggingface/small_things

2024-07-25 20:29:13 -07:00

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_v2.py

Add a few missing __all__ entries.

2024-08-07 09:35:51 -07:00

efficientformer.py

…

efficientnet.py

Add mobilenet_edgetpu_v2_m weights

2024-07-28 17:19:36 -07:00

efficientvit_mit.py

…

efficientvit_msra.py

…

eva.py

Merge pull request #2225 from huggingface/small_things

2024-07-25 20:29:13 -07:00

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

Add a few missing __all__ entries.

2024-08-07 09:35:51 -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

Merge pull request #2225 from huggingface/small_things

2024-07-25 20:29:13 -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

…

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

Merge pull request #2225 from huggingface/small_things

2024-07-25 20:29:13 -07:00

resnetv2.py

…

rexnet.py

…

selecsls.py

…

senet.py

…

sequencer.py

…

sknet.py

…

swin_transformer_v2_cr.py

Merge pull request #2225 from huggingface/small_things

2024-07-25 20:29:13 -07:00

swin_transformer_v2.py

Merge pull request #2225 from huggingface/small_things

2024-07-25 20:29:13 -07:00

swin_transformer.py

Merge pull request #2225 from huggingface/small_things

2024-07-25 20:29:13 -07:00

tiny_vit.py

…

tnt.py

…

tresnet.py

…

twins.py

…

vgg.py

…

visformer.py

…

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

vision_transformer.py

Merge pull request #2225 from huggingface/small_things

2024-07-25 20:29:13 -07:00

vitamin.py

…

volo.py

…

vovnet.py

…

xception_aligned.py

…

xception.py

…

xcit.py

…