Laden und Prüfen von trainierten Modellen
Du baust einen KI-Assistenten auf, der einen menschenähnlichen Dialog über eine Vielzahl von Themen führen kann. Dabei nutzt du das leistungsstarke BERT-Modell, das auf einem großen Korpus von Textdaten trainiert wurde.
Du druckst die Konfiguration aus, um zu überprüfen, ob du ein KI-Konversationsmodell mit bestimmten Parametern wie model_type geladen hast: bert, num_attention_heads: 12, und num_hidden_layers: 12.
Diese Übung ist Teil des Kurses
Effizientes KI-Modelltraining mit PyTorch
Anleitung zur Übung
- Initialisiere die Modellparameter mit der entsprechenden AutoModel-Klasse, um das Modell
bert-base-uncasedzu laden. - Drucke die Konfiguration des Modells.
Interaktive Übung
Vervollständige den Beispielcode, um diese Übung erfolgreich abzuschließen.
from transformers import AutoModelForSequenceClassification
# Load a pre-trained bert-base-uncased model
model = ____.____("bert-base-uncased")
# Print the model's configuration
print(model.____)