사전 학습된 모델 불러오기 및 검사하기
다양한 주제에 걸쳐 자연스러운 대화가 가능한 대화형 AI 어시스턴트를 만들고 있습니다. 대규모 텍스트 데이터로 사전 학습된 강력한 BERT 모델을 활용할 예정입니다.
모델 구성을 출력해 model_type: bert, num_attention_heads: 12, num_hidden_layers: 12 등의 파라미터가 올바르게 설정된 대화형 AI 모델을 불러왔는지 확인해 보세요.
이 연습은 강의의 일부입니다
PyTorch로 AI 모델 효율적으로 학습시키기
연습 안내
- 적절한 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.____)