pytorch报错Unable to get repr for

最新推荐文章于 2023-12-25 09:07:23 发布

AI算法网奇

最新推荐文章于 2023-12-25 09:07:23 发布

阅读量8.2k

点赞数 1

分类专栏： pytorch知识宝典

本文链接：https://blog.csdn.net/jacke121/article/details/108986007

版权

pytorch知识宝典专栏收录该内容

503 篇文章 228 订阅 ¥29.90 ¥99.00

订阅专栏

超级会员免费看

在使用PyTorch进行调试时，遇到'Unable to get repr for <class 'torch.Tensor'>'错误，通常与数据格式有关。该问题在执行RoIPooling操作时出现，由于RoIs数据类型不正确导致。解决方案是使用`torch.tensor()`来创建Tensor，而非`torch.from_numpy()`或`torch.FloatTensor()`，尽管它们的数值相同，但可能导致错误。

摘要由CSDN通过智能技术生成

pytorch报错Unable to get repr for

标签出现了-1,2或者其他值，loss计算之后，

报错：CUDA error: device-side assert triggered

然后出现此错误：

Unable to get repr for class tensor

pytorch 在调试的时候，变量出现了Unable to get repr for <class ‘torch.Tensor’>

是自己数据格式的问题，具体原因后面在详细整理。

具体情景： 做roipooling的时候，rois的数据类型不对，出现了以上错误。
解决方法：采用torch.tensor([0, 2, 2, 10, 10])，就是采用了torch.tensor。前期有尝试torch.from_numpy， torch.FloatTensor()，虽然数据值都是一样的，但是就是报错。

具体代码如下：

temp = rois.numpy().tolist()
# rois = torch.FloatTensor([0] + [temp[0][1], temp[0][2], temp[0][3], temp[0][4]] )
rois = torch.tensor([0] + [temp[0][1], temp[0][2], temp[0][3], temp[0][4]])
pooled_feat = pooling(base_feat, rois.view(-1, 5))