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

文章来源: blog.csdn.net，作者：网奇，版权归原作者所有，如需转载，请联系作者。

原文链接：blog.csdn.net/jacke121/article/details/108986007