pytorch报错Unable to get repr for

最新推荐文章于 2025-02-09 14:25:06 发布

原创最新推荐文章于 2025-02-09 14:25:06 发布 · 8.9k 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#pytorch

pytorch 专栏收录该内容

80 篇文章

订阅专栏

博客主要讲述了Pytorch调试时变量出现Unable to get repr for <class ‘torch.Tensor’>的问题。经分析是数据格式问题，如做roipooling时rois数据类型不对。解决方法是采用torch.tensor，前期尝试torch.from_numpy等方法均报错。

文章目录

- - pytorch 在调试的时候，变量出现了Unable to get repr for

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))