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(____)