Завантаження та огляд попередньо натренованих моделей
Ви створюєте розмовного помічника ШІ, який може підтримувати діалог у людському стилі на широкий спектр тем, використовуючи потужну модель BERT, попередньо натреновану на великому корпусі текстових даних.
Ви виведете конфігурацію, щоб перевірити, що завантажено розмовну модель ШІ з певними параметрами, як-от model_type: bert, num_attention_heads: 12 і num_hidden_layers: 12.
Ця вправа є частиною курсу
Ефективне тренування моделей ШІ з PyTorch
Інструкції до вправи
- Ініціалізуйте параметри моделі за допомогою відповідного класу AutoModel, щоб завантажити модель
bert-base-uncased. - Виведіть конфігурацію моделі.
Інтерактивна практична вправа
Спробуйте виконати цю вправу, доповнивши цей зразок коду.
from transformers import AutoModelForSequenceClassification
# Load a pre-trained bert-base-uncased model
model = ____.____("bert-base-uncased")
# Print the model's configuration
print(model.____)