pytorch报错集

11 RuntimeError: some of the strides of a given numpy array are negative. This is currently not supported, but will be added in future releases.

这个原因是因为程序中操作的numpy中有使用负索引的情况:image[…, ::-1]。

解决办法比较简单,加入image这个numpy变量引发了错误,返回image.copy()即可。因为copy操作可以在原先的numpy变量中创造一个新的不适用负索引的numpy变量。

RuntimeError: “upsample_nearest2d” not implemented for ‘Byte’

img = cv2.imread(img_path)
img = img[:,:,::-1].copy()
img = torch.from_numpy(img)
img = img.permute(2,0,1)
img = pad_to_square(img, 0)
print(img.shape, img.unsqueeze(0).shape, img.dtype)
img = F.interpolate(img.unsqueeze(0), size=self.size, mode='nearest')

原因是torch.from_numpy默认为torch.uint8,转换为FloatTensor类型(或者其他)就好。

img = torch.from_numpy(img).float()

Broken pipe

原因是Dataloader在windows10下不支持多线程,把dataloader的num_workers参数改为0即可。

RuntimeError: CUDA error: device-side assert triggered void cunn_ClassNLLCriterion_updateOutput_k

转载:https://blog.csdn.net/weixin_37142859/article/details/93844144
使用pytorch的时候报这个错误说明你label中有些指不在[0, num classes), 区间左闭右开。比如类别数num_class=3, 你的label出现了-1或者3, 4, 5等!!!!

你可能感兴趣的:(pytorch报错集)