使用Pytorch实现Transformer 对应的博客地址为:https://blog.csdn.net/ing100/article/details/132979122?spm=1001.2014.3001.5502 使用的Pytorch版本为1.11.0 本文实现了Transformer并使用三句话作为训练集进行训练 运行代码 运行 main.py 就会开始训练+预测,若只想进行预测就在代码末尾将train()函数注释即可