Pytorch TypeError：eq（）收到了无效的参数组合

作者: LOGO
发布时间: 2024-08-13 11:46:29 (3天前)
转自：

2 条回复

0#
回复此人
一生浮华 | 2019-08-31 10-32

<div class =“post-text”itemprop =“text”> <P> 你想要比较 <code> predicted </code> 和 <code> labels </code> 。但是，你的 <code> predicted </code> 是一个 <code> np.array </code> 而 <code> labels </code> 是一个 <code> torch.tensor </code> 因此 <code> eq() </code> （该 <code> == </code> operator）无法比较它们。 <BR/> 更换 <code> np.argmax </code> 同 <a href="https://pytorch.org/docs/0.4.1/torch.html#torch.argmax" rel="nofollow noreferrer"> <code> torch.argmax </code> </A> ： </p> <pre> <code> return torch.argmax(mean, dim=1) </code> </pre> <P> 你应该没问题。 </p> </DIV>

编辑

登录后才能参与评论