開始使用免費開始

PyTorch Transformer

現在你已經熟悉 Transformer 架構的各個元件了,來親自定義一個吧!torch.nn 模組(已替你匯入為 nn)只要幾行程式碼就能優雅地完成這件事。

本練習屬於課程

Transformer Models with PyTorch

檢視課程

練習說明

  • 定義一個具有 8 個注意力頭、6 個 encoder 與 6 個 decoder 層,且輸入序列嵌入向量(embedding)長度為 1536 的 transformer。
  • 印出模型物件以檢視模型架構。

動手互動練習

試著完成這個範例程式碼,體驗一下這個練習。

# Define the transformer model
model = ____

# Print the model object
print(____)
編輯並執行程式碼