Încărcarea și inspectarea modelelor pre-antrenate
Construiești un asistent conversațional bazat pe inteligență artificială, capabil să poarte dialoguri naturale pe o gamă largă de subiecte, folosind puternicul model BERT pre-antrenat pe un corpus extins de date text.
Vei afișa configurația pentru a verifica că ai încărcat un model de AI conversațional cu anumiți parametri, cum ar fi model_type: bert, num_attention_heads: 12 și num_hidden_layers: 12.
Acest exercițiu face parte din cursul
Antrenament eficient al modelelor AI cu PyTorch
Instrucțiuni pentru exercițiu
- Inițializează parametrii modelului cu clasa AutoModel potrivită pentru a încărca modelul
bert-base-uncased. - Afișează configurația modelului.
Exercițiu interactiv practic
Încearcă acest exercițiu completând acest cod de exemplu.
from transformers import AutoModelForSequenceClassification
# Load a pre-trained bert-base-uncased model
model = ____.____("bert-base-uncased")
# Print the model's configuration
print(model.____)