weixin_40740309的博客

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

事情是这样的，SVHN数据集是一个三通道的RGB数据集，想把它转为单通道的灰度图，你在网上搜索大家都是下面这样使用的： train_dataset = SVHN('./data', split='train', download=True, transform=transforms.Compose([ transforms.Resize(28), transforms.Grayscale(num_output_c

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

踩坑 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