想当厨子的半吊子程序员-CSDN博客

原创 ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1, 256,

pytorch在对以经训练好的网络进行测试时发现以上问题，在装载网络后添加model.eval（），问题得到解决。model = nn.DataParallel(model).cpu()model.load_state_dict(torch.load(path, map_location=torch.device('cpu')), False)model.eval()此代码段有三个需要注意的地方 nn.DataParallel(model)如果在训练网络时加入了这个函数，在使用训练好

2021-03-04 21:39:18 573 1

原创 pytorch学习之合并与拆分Cat、stack、split、chunk

1、cat函数a = torch.rand(4,32,8)b = torch.rand(5,32,8)print(torch.cat([a,b],dim=0).shape) # torch.Size([9, 32, 8])cat函数的第一个参数为需要合并的两个张量，第二个参数dim表示哪一维需要合并，如上式表示第0维需要合并，注：除了需合并的维数上的数目可以不同，其他维数需相同2、stack函数a = torch.rand(32,8)b = torch.ran

2020-10-25 11:31:24 931

原创 pytorch学习之维度变化

1、view函数：a = torch.rand(4,1,28,28)b = a.view(4,28*28) print(b.shape) #result:torch.Size([4, 784])首先产生一个维数为4*1*28*28的张量a，通过view函数对a张量进行操作，将a张量变为一个4*[28*28]的张量，改变之后的张量结果为上式，需注意，改变前后的元素个数不变，即4*1*28*28需等于4*[28*28]，例如可以对1*28*28的照片进行处理，

2020-10-23 17:24:33 855

原创 pytorch学习笔记之tensor的创建与切片

直接粘贴代码，其中创建tensor的创建方法不全，有需要的查找，每一行为一个创建方法，每行的第二个#号之后print函数内的操作为对tensor的操作，之后为输出结果，最后为操作的粗略解释import torch# a = torch.randn(3,4) #创建一个3*4的张量，其中元素为随机的正太数# a = torch.full([2,2],2) #生成2*2，元素值都为2的张量# b = torch.Tensor(2,2)# a = torch.arange(0,10,

2020-10-21 21:58:21 642

原创 Pytorch学习笔记之通过numpy实现线性拟合

通过使用numpy库编写简单的Gradient Descent数据位于附件之中import torchfrom torch import autogradimport numpy as npimport matplotlib.pyplot as plt'''torch关于求导的简单运用'''# x = torch.tensor(1.)# a = torch.tensor(1.,requires_grad=True)# b = torch.tensor(2.,requires_grad=T

2020-10-21 10:27:26 657

原创总结matlab2018a安装后遇到matlab license manager error -8

安装教程和破解教程参考此博主文章：安装 | MATLAB2018a (64位) 安装教程及安装包下载链接在安装完成之后，按照教程将path文件中的netapi32.dll文件放入安装目录bin/win64/ , license_standalone.lic文件放入license文件夹下之后，回到桌面打开matlab显示 matlab license manager error -8 的错误经过查找之后发现两种解决办法：1：检查系统是否为最新系统，升级一下电脑系统进行尝试（我的电脑尝...

2020-09-07 08:03:26 14562 66

原创 Python中return与yield的区别

Python中return与yield的区别**相同点****不同点**在学习scrapy中遇到在spider中处理数据之后使用yield返回结果，而不是return返回结果，有所疑问，故而baidu得到以下结果。相同点return与yield相同，都可以返回结果不同点按照定义定义来说yield是让函数成为一个生成器，在我自己理解的话，所谓生成器，就比方说得到一个结果则返回一个结果...

2019-07-15 16:08:36 434

原创 Linux学习笔记

linux指令学习笔记linux指令使用格式对于一般的linux指令来说，使用格式一般为指令参数选项（对于参数与选项来说，可根据不同的需求选择不同的参数，参数即我们通常看到的-a,-c等）对于某个指令的参数来说，可以通过指令 --help来查看不同参数的功能个人认为对于新人首要需要知道的命令是 cd使用linux终端首先要进行的是进入某个文件夹这样一来就涉及到了cd命令如果要进入当...

2018-11-12 10:50:20 133

qq_42191909的博客