- 博客(4)
- 收藏
- 关注
RSS订阅原创 RuntimeError: The size of tensor a (128) must match the size of tensor b (16) at non-singleton dimen
RuntimeError: The size of tensor a (128) must match the size of tensor b (16) at non-singleton dimension 0这个错误出现的原因有好多种！！其中一个比较可能的原因是因为：封装dataloader的时候，最后剩下的不足一个batchsize！ 自带的dataloader就会有这个现象。改正：batch_size_s = len(targets) #不足一个batch_size直接停止训练if bat
2021-04-04 20:25:13
7309
5
原创 踩坑 SVHN数据集transforms不起作用
事情是这样的，SVHN数据集是一个三通道的RGB数据集，想把它转为单通道的灰度图，你在网上搜索大家都是下面这样使用的： train_dataset = SVHN('./data', split='train', download=True, transform=transforms.Compose([ transforms.Resize(28), transforms.Grayscale(num_output_c
2021-03-12 17:21:32
953
2
原创 TypeError: Cannot handle this data type: (1, 1, 28), |u1
PIL和numpy相互转换方式如下：from PIL import Image#PIL image转成numpynp_img = np.asarray(PIL_img) #参数时PIL类型的图片或 np_img = np.array(PIL_img)#将array准成PIL imageImage.fromarray(np.uint8(np_img))但是直接用Image.fromarray(np.uint8(data))转换会报错：TypeError: Cannot handle t
2021-03-12 17:18:34
16423
18
原创 踩坑 ValueError：only one element tensors can be converted to Python scalarsC
踩坑 ValueError：only one element tensors can be converted to Python scalarsClist 转 torch.Tensortensor=torch.Tensor(list)注意：有时，上面操作会出现报错：ValueError:only one element tensors can be converted to Python scalars原因是：要转换的list里面的元素包含多维的tensor。就像这个样子 [tensor，t
2021-03-12 16:56:52
18380
9
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人