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}