Začněte nyníZačněte zdarma

Načítání a prozkoumávání předtrénovaných modelů

Vytváříš konverzačního AI asistenta, který dokáže vést přirozený dialog na nejrůznější témata. Využívá přitom výkonný model BERT předtrénovaný na rozsáhlém textovém korpusu.

Vypíš konfiguraci modelu a ověř, že jsi načetl/a konverzační AI model se správnými parametry, jako jsou model_type: bert, num_attention_heads: 12 a num_hidden_layers: 12.

Toto cvičení je součástí kurzu

Efficient AI Model Training with PyTorch

Zobrazit kurz

Pokyny k cvičení

  • Inicializuj parametry modelu pomocí příslušné třídy AutoModel a načti model bert-base-uncased.
  • Vypiš konfiguraci modelu.

Interaktivní cvičení na vyzkoušení si v praxi

Vyzkoušejte si toto cvičení dokončením tohoto ukázkového kódu.

from transformers import AutoModelForSequenceClassification

# Load a pre-trained bert-base-uncased model
model = ____.____("bert-base-uncased")

# Print the model's configuration
print(model.____)
Upravit a spustit kód