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

演習

事前学習済み PyTorch モデルでの言語翻訳

PyBooks のチームでは、ある言語から別の言語への翻訳を行う AI プロジェクトに取り組んでいます。このタスクには事前学習済みモデルを活用したいと考えており、これにより学習時間とリソースを大幅に節約できます。この演習では、HuggingFace の Transformers ライブラリから T5（Text-To-Text Transfer Transformer）モデルをセットアップし、英語のフレーズをフランス語に翻訳します。

T5Tokenizer、T5ForConditionalGeneration はあらかじめ読み込まれています。

指示

100 XP
  • 事前学習済みの "t5-small" から tokenizer と model を初期化します。
  • tokenizer を使って入力プロンプトをエンコードし、PyTorch のテンソルを返すようにします。
  • model を用いて入力プロンプトを翻訳し、翻訳結果を生成します。