RuntimeError: Found dtype Double but expected Float

[scb-vs5]

想问一下执行的时候，scaler.scale(loss).backward会报错，Found dtype Double but expected Float，在stackoverflow上面查找并尝试了多种方式，但是一直没能解决，想请教一下这个该怎么处理？

[scb-vs5]

子loss和总loss我看到都是torch.float32了的，但是很奇怪它还是会报错，不知道是不是环境的问题？我用的是pytorch1.8，请问有什么思路嘛

[scb-vs5]

跑通，换成更高版本的torch（我的是1.12.1）再加调试一下即可

[Felix-fz]

请问你是直接在他这个代码基础上跑通的吗，有没有做修改