RuntimeError: Expected object of

2019-03-21  本文已影响0人  量产机

找到报错行

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}

上一篇 下一篇

猜你喜欢

热点阅读