Pytorch官方Transformer教程沒有使用nn.Transformer,僅僅使用了TransformerEncoderLayer。甚至連TransformerDecoderLayer都沒有用。我找到了一個nn.Transformer示例代碼,自己改了改放在github上:https://github.com/Kenneth111/TransformerDemo
有興趣的可以參考一下。
Pytorch官方Transformer教程沒有使用nn.Transformer,僅僅使用了TransformerEncoderLayer。甚至連TransformerDecoderLayer都沒有用。我找到了一個nn.Transformer示例代碼,自己改了改放在github上:https://github.com/Kenneth111/TransformerDemo
有興趣的可以參考一下。