始める無料で始める

PyTorch の Transformer

すでにトランスフォーマーアーキテクチャの各コンポーネントに慣れてきましたので、いよいよモデルを定義してみましょう! torch.nn モジュール(ここでは nn としてインポート済み)を使えば、わずか数行で実装できます。

この演習はコースの一部です

PyTorchで学ぶTransformerモデル

コースを見る

演習の手順

  • アテンションヘッドを 8、エンコーダ層とデコーダ層をそれぞれ 6、入力系列の埋め込み長を 1536 にして、Transformer を定義します。
  • モデルオブジェクトを出力して、モデルのアーキテクチャを確認します。

実践的なインタラクティブ演習

このサンプルコードを完成させて、この演習に挑戦してみましょう。

# Define the transformer model
model = ____

# Print the model object
print(____)
コードを編集して実行