2024年12月14日
Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same
摘要： 错误报在了forward里的Conv2d处。原因是函数写在forward里可能默认cpu，如果写在init构造函数里，就不需要再指定cuda。 修改为箭头指示就不再报错了。 【参考】 Input type (torch.cuda.FloatTensor) and weight type (torch 阅读全文
posted @ 2024-12-14 13:27 夕西行 阅读(12) 评论(0) 推荐(0) 编辑