1. 学ぶ
  2. /
  3. コース
  4. /
  5. PyTorch による効率的な AI モデルトレーニング

Connected

演習

事前学習済みモデルの読み込みと確認

幅広いトピックで人間らしい対話ができる会話型 AI アシスタントを構築しましょう。ここでは、大量のテキストデータで事前学習された強力な BERT モデルを活用します。

設定情報を出力して、model_type: bert、num_attention_heads: 12、num_hidden_layers: 12 といったパラメータを持つ会話型 AI モデルが正しく読み込まれているか確認しましょう。

指示

100 XP
  • 適切な AutoModel クラスを使って bert-base-uncased モデルのパラメータを初期化します。
  • モデルの設定情報を出力します。