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
有兴趣的可以参考一下。