找到报错行
label = {'img':self.normalize(tensor_img), 'dis': sp_pdf, 'prrs': pr_vec}
将tensor_img改为tensor_img.float()即可
label = {'img':self.normalize(tensor_img.float()), 'dis': sp_pdf, 'prrs': pr_vec}
RuntimeError: Expected object of type torch.DoubleTensor but found type torch.FloatTensor
©著作权归作者所有,转载或内容合作请联系作者
平台声明：文章内容（如有图片或视频亦包括在内）由作者上传并发布，文章内容仅代表作者本人观点，简书系信息发布平台，仅提供信息存储服务。
平台声明：文章内容（如有图片或视频亦包括在内）由作者上传并发布，文章内容仅代表作者本人观点，简书系信息发布平台，仅提供信息存储服务。
推荐阅读更多精彩内容
- 该文章为转载文章，作者简介：汪剑，现在在出门问问负责推荐与个性化。曾在微软雅虎工作，从事过搜索和推荐相关工作。 T...
- 本文链接：https://blog.csdn.net/thesby/article/details/5126443...
- 《÷》既能令乐迷听到熟悉的情感民谣，也有基于Hip-pop节拍之上赋予激情的Rap，以及听到无数次吉他清弹出的杰作...
