pytorch
YYYYY_Z
这个作者很懒,什么都没留下…
展开
-
pytorch中的model.eval() 和model.train()以及with torch.no_grad 还有torch.set_grad_enabled总结
一、pytorch中的model.eval() 和 model.train()再pytorch中我们可以使用eval和train来控制模型是出于验证还是训练模式,那么两者对网络模型的具体影响是什么呢?1. model.eval()eval主要是用来影响网络中的dropout层和batchnorm层的行为。在dropout层保留所有的神经网络单元,batchnorm层使用在训练阶段学习得到的mean和var值。另外eval不会影响网络参数的梯度的计算,只不过不回传更新参数而已。所以eval模式要比wit原创 2020-09-15 10:49:44 · 7955 阅读 · 4 评论 -
basemap中出现IndexError: only integers, slices (`:`), ellipsis (`...`), numpy.newaxis (`None`
具体的出错信息为:Traceback (most recent call last): File "basemap.py", line 32, in <module> m.contourf(x, y, data, clevs, colors=color) File "/home/zhaoyy/miniconda3/envs/2020huawei/lib/python3.5/site-packages/mpl_原创 2020-06-11 23:36:39 · 1921 阅读 · 4 评论 -
Segmentation fault (core dumped)
使用pytorch进行深度学习模型训练时,莫名出现了这个错误。之前用的好好的。出现这个错误的原因有很多,网上与很多人说是linux中对栈的大小做了限制,具体的大家可以对照这篇blog进行设置。但我遇到不是这种情况,我是在导入matplotlib是出现了这个错误,于是参考了这篇blog。import matplotlib as mplmpl.use('TkAgg') # or whatever other backend that you wantimport matplotlib.pyplot原创 2020-05-10 22:47:02 · 1116 阅读 · 1 评论 -
pytorch中的transpose方法(函数)
pytorch中的transpose方法的作用是交换矩阵的两个维度,transpose(dim0, dim1) → Tensor,其和torch.transpose()函数作用一样。torch.transpose():torch.transpose(input, dim0, dim1) → TensorReturns a tensor that is a transposed versio...原创 2019-10-19 11:30:36 · 53125 阅读 · 3 评论