Mulai sekarangMulai gratis

Memuat dan memeriksa model pralatih

Anda sedang membangun asisten AI percakapan yang dapat berdialog layaknya manusia di berbagai topik, dengan memanfaatkan model BERT yang kuat dan telah dilatih sebelumnya pada korpus data teks yang besar.

Anda akan mencetak konfigurasi untuk memverifikasi bahwa Anda telah memuat model AI percakapan dengan parameter tertentu seperti model_type: bert, num_attention_heads: 12, dan num_hidden_layers: 12.

Latihan ini merupakan bagian dari kursus

Pelatihan Model AI Efisien dengan PyTorch

Lihat Kursus

Instruksi latihan

  • Inisialisasi parameter model dengan kelas AutoModel yang sesuai untuk memuat model bert-base-uncased.
  • Cetak konfigurasi model.

Latihan interaktif langsung praktik

Cobalah latihan ini dengan melengkapi kode contoh ini.

from transformers import AutoModelForSequenceClassification

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

# Print the model's configuration
print(model.____)
Edit dan Jalankan Kode