2025年2月24日
RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same
摘要： 练过程中出现如题错误的解决方案 常规解决方案从报错问题描述中可以找到错误原因 输入的数据类型为torch.cuda.FloatTensor，说明输入数据在GPU中模型参数的数据类型为torch.FloatTensor，说明模型还在CPU问题原因搞清楚了，模型没加载到CPU，在代码中加一行语句就可以了 阅读全文
posted @ 2025-02-24 20:59 视觉书虫 阅读(934) 评论(0) 推荐(0)