事前学習済みGPT-2モデルでのテキスト補完
PyBooksに戻りましょう。現在のプロジェクトでは、既存の物語をもとに魅力的なストーリーを生成し、顧客を惹きつけて体験を高めることが目標です。そのためには、自然に続きが書ける強力なテキスト生成ツールが必要です。ここでは事前学習済みモデルを使って実現します。
次のライブラリは読み込まれています:torch と、transformers からの GPT2Tokenizer、GPT2LMHeadModel。
この演習はコースの一部です
PyTorch で学ぶテキストの Deep Learning
実践的なインタラクティブ演習
このサンプルコードを完成させて、この演習に挑戦してみましょう。
# Initialize the tokenizer
tokenizer = ____
# Initialize the pre-trained model
model = ____