好像不支持AMP训练，应如何修改？

[MMagicLoren]

RuntimeError: Input type (torch.cuda.HalfTensor) and weight type (torch.cuda.FloatTensor) should be the same

[xyl-507]

我没遇见这种情况。我看网上有教程，你可以试试 https://blog.csdn.net/qq_35759272/article/details/137009191

…

---- 回复的原邮件 ---- | 发件人 | zhenjiang ***@***.***> | | 日期 | 2025年04月18日 22:39 | | 收件人 | ***@***.***> | | 抄送至 | ***@***.***> | | 主题 | [xyl-507/SmallTrack] 好像不支持AMP训练，应如何修改？ (Issue #12) | RuntimeError: Input type (torch.cuda.HalfTensor) and weight type (torch.cuda.FloatTensor) should be the same — Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you are subscribed to this thread.Message ID: ***@***.***> MMagicLoren created an issue (xyl-507/SmallTrack#12) RuntimeError: Input type (torch.cuda.HalfTensor) and weight type (torch.cuda.FloatTensor) should be the same — Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you are subscribed to this thread.Message ID: ***@***.***>

[zhen8421]

RuntimeError: Input type (torch.cuda.HalfTensor) and weight type (torch.cuda.FloatTensor) should be the same

你好，你修改成功了吗