- 博客(6)
- 收藏
- 关注
RSS订阅原创 [已解决]No module named ‘torchsummary‘
报错如下原因分析torchsummary 可以做Pytorch可视化，输出网络相关信息。当前环境缺失torchsummary安装包。解决方案pip install torchsummary注：conda install torchsummary可能无法安装，那就直接用pip吧。搞定！...
2021-04-25 16:00:52
9053
3
原创 【已解决】RuntimeError: stack expects each tensor to be equal size, but got [3, 256, 341] at entry 0
【报错】如下：RuntimeError: stack expects each tensor to be equal size, but got [3,256, 341] at entry 0 and [3, 256, 386] at entry 1【分析】train_dataset = datasets.ImageFolder( traindir, transforms.Compose([ transforms.Resize((224)) ###【原因】原
2021-04-21 11:54:16
25706
22
原创 Jupyter Notebook操作快捷键【持续增加】
1.Jupyter Notebook默认不显示行号，可是当我们代码报错时，发现会显示自己多少行出现错误快捷键shift + L搞定！
2021-04-16 16:46:53
164
原创 PyTorch保存、加载模型【简单暴力】
pytorch保存模型的方式有两种：将整个网络都都保存下来仅保存和加载模型参数（推荐使用这样的方法）代码部分方法1# 保存和加载整个模型torch.save(model_object, 'model.pkl')model = torch.load('model.pkl')方法2# 仅保存和加载模型参数(推荐使用)torch.save(model_object.state_dict(), 'params.pkl')model_object.load_state_dict.
2021-04-14 21:25:07
264
原创 【已解决】NameError: name ‘xrange‘ is not defined
Fork人家代码的时候，报错如下：原因：xrange( )函数时在python 2.x中的一个函数，在Python3中，range()的实现方式与xrange()函数相同，所以就不存在专用的xrange( )。解决方案：如继续在python 3中运行程序，直接将xrange( )函数全部换为range( )即可！============================ 分割线 ===================================以下补充介绍xrange()的用法，pyth
2021-04-08 10:52:28
658
原创 Pytorch中squeeze()函数
squeeze() 函数：从数组的形状中删除单维度条目，即把shape中为1的维度去掉用法：numpy.squeeze(a,axis = None)a表示输入的数组；axis用于指定需要删除的维度，但是指定的维度必须为单维度，否则将会报错；axis的取值可为None 或 int 或 tuple of ints, 可选。若axis为空，则删除所有单维度的条目；返回值：数组；不会修改原数组； >>> a = e.reshape(1,1,10) >>
2021-04-07 16:04:23
1385
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人