1.需要再main.py里的在DataLoader()中添加一个参数generator=torch.Generator(device=‘cuda’),这样就可以在GPU上创建一个随机数生成器,而不是默认的CPU上的生成器。 2.需要将position_embedding.py的view()函数换成reshape(),这是在colab的情况下。 3.data是需要自己创建文件夹,下载并解压,然后在main.py的–dataset参数中修改默认路径。