1. 学ぶ
  2. /
  3. コース
  4. /
  5. PyTorch で学ぶテキストの Deep Learning

Connected

演習

事前学習済みGPT-2モデルでのテキスト補完

PyBooksに戻りましょう。現在のプロジェクトでは、既存の物語をもとに魅力的なストーリーを生成し、顧客を惹きつけて体験を高めることが目標です。そのためには、自然に続きが書ける強力なテキスト生成ツールが必要です。ここでは事前学習済みモデルを使って実現します。

次のライブラリは読み込まれています:torch と、transformers からの GPT2Tokenizer、GPT2LMHeadModel。

指示1 / 3

undefined XP
    1
    2
    3
  • gpt2 用の事前学習済みGPT-2トークナイザーを初期化します。
  • gpt2 用の事前学習済みGPT-2のLMHeadモデルを初期化します。