timm documentation

BYOANet (BoTNet, HaloNet, LambdaNet)

You are viewing main version, which requires installation from source. If you'd like regular pip install, checkout the latest stable version (v1.0.31).
Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

BYOANet (BoTNet, HaloNet, LambdaNet)

Catégorie : Hybride · Code : timm/models/byoanet.py · Architectures : 15 · Poids pré-entraînés : 13 (13 avec des résultats ImageNet-1k)

Les réseaux construits avec le système de configuration de blocs / étages ByobNet qui placent des couches d'auto-attention (ou similaires) dans des étages de type ResNet : Bottleneck Transformers (BoTNet), HaloNets et Lambda ResNets. La docstring du module marque ces définitions comme expérimentales.

Articles

Utilisation

Consultez le Démarrage rapide pour le chargement des poids pré-entraînés, le prétraitement des images et l’ajustement fin, ainsi que Extraction de caractéristiques pour utiliser les modèles comme squelettes.

>>> import timm
>>> model = timm.create_model('halonet50ts.a1h_in1k', pretrained=True)
>>> data_config = timm.data.resolve_model_data_config(model)

Prise en charge de l'API des modèles

FonctionnalitéByobNet
cartes de caractéristiques features_only=True✓ (pas 2, 4, 8, 16, 32)
forward_intermediates()✓
prune_intermediate_layers()✓
Toute taille d'entrée, aucune modification nécessaire✗
set_input_size()✗
dynamic_img_size=True✗
set_grad_checkpointing()✓
Décroissance du taux d'apprentissage par couche (group_matcher())✓
reset_classifier()✓

✓ pris en charge, ✓ (carré) fonctionne uniquement pour les tailles carrées, ✗ non pris en charge, – non nécessaire (le modèle accepte déjà n'importe quelle taille d'entrée). Vérifié en exécutant chaque fonctionnalité sur un modèle initialisé aléatoirement (ByobNet avec halonet_h1 à 256), les autres architectures de la même classe sont censées se comporter de la même manière. Consultez la page API du modèle pour plus de détails.

Architectures

ArchitectureParamètres (M)GMACsTaille d'entréePoids pré-entraînés
botnet26t_25612.493.322561
botnet50ts_25622.745.542560
eca_botnext26ts_25610.592.462561
eca_halonext26ts10.762.442561
halo2botnet50ts_25622.645.022561
halonet26t12.483.192561
halonet50ts22.735.302561
halonet_h18.103.002560
haloregnetz_b11.681.972241
lambda_resnet26rpt_25610.993.162561
lambda_resnet26t10.963.022561
lambda_resnet50ts21.545.072561
lamhalobotnet50ts_25622.575.022561
sebotnet33ts_25613.703.892561
sehalonet33ts13.693.552561

Les GMACs et le nombre de paramètres correspondent à la taille d’entrée par défaut et au classifieur de chaque architecture, d’après les résultats des benchmarks.

Poids pré-entraînés

Top-1 est la précision de validation sur ImageNet-1k issue de results-imagenet.csv à la taille d'évaluation indiquée, - si le poids n'a pas de résultat ImageNet-1k (par exemple, poids de pré-entraînement ou poids image-texte). Classes est la taille du classifieur pré-entraîné, aucune pour les poids sans classifieur.

Les poids sont regroupés par source : le papier ou l’origine indiqué dans la configuration pré-entraînée, à défaut une organisation du Hub autre que timm, à défaut la première partie de l’étiquette pré-entraînée (architecture.tag). Les sources ne comportant qu’un seul poids sont listées sous other.

balises a1h

PoidsTaille d'entréeTop-1 (%)Taille d'évaluationClassesLicenceHub
halo2botnet50ts_256.a1h_in1k25682.0842561000apache-2.0timm
halonet26t.a1h_in1k25679.1342561000apache-2.0timm
halonet50ts.a1h_in1k25681.6402561000apache-2.0timm
lambda_resnet50ts.a1h_in1k25681.1842561000apache-2.0timm
lamhalobotnet50ts_256.a1h_in1k25681.5522561000apache-2.0timm
sebotnet33ts_256.a1h_in1k25681.1562561000apache-2.0timm

c1 tags

PoidsTaille d'entréeTop-1 (%)Taille d'évaluationClassesLicenceHub
botnet26t_256.c1_in1k25679.2562561000apache-2.0timm
eca_botnext26ts_256.c1_in1k25679.2622561000apache-2.0timm
eca_halonext26ts.c1_in1k25679.5182561000apache-2.0timm
lambda_resnet26rpt_256.c1_in1k25678.9602561000apache-2.0timm
lambda_resnet26t.c1_in1k25679.1262561000apache-2.0timm

Autres

PoidsTaille d'entréeTop-1 (%)Taille d'évaluationClassesLicenceHub
haloregnetz_b.ra3_in1k22481.0582241000apache-2.0timm
sehalonet33ts.ra2_in1k25680.9762561000apache-2.0timm
Update on GitHub