Tải và kiểm tra mô hình đã huấn luyện sẵn
Bạn đang xây dựng một trợ lý AI hội thoại có thể trò chuyện tự nhiên về nhiều chủ đề, tận dụng sức mạnh của mô hình BERT đã được huấn luyện sẵn trên một tập văn bản lớn.
Bạn sẽ in cấu hình để xác nhận rằng bạn đã tải một mô hình AI hội thoại với các tham số như model_type: bert, num_attention_heads: 12, và num_hidden_layers: 12.
Bài tập này là một phần của khóa học
Huấn luyện Mô hình AI Hiệu quả với PyTorch
Hướng dẫn bài tập
- Khởi tạo tham số mô hình bằng AutoModel phù hợp để tải mô hình
bert-base-uncased. - In cấu hình của mô hình.
Bài tập tương tác thực hành trực tiếp
Hãy thử làm bài tập này bằng cách hoàn thành đoạn mã mẫu này.
from transformers import AutoModelForSequenceClassification
# Load a pre-trained bert-base-uncased model
model = ____.____("bert-base-uncased")
# Print the model's configuration
print(model.____)