- 博客(9)
- 收藏
- 关注
原创 pytorch nn.Module train和eval 函数 深入解析
1、先看看官方的解释模型调用eval() 就是设置为评估模式,调用train(True)就为训练模式,这个说的很不透彻,我们一起来探究究竟为何model模型就设置为了评估模式2、深度解析先说结论:模型类继承了nn.Module 就有实例属性traing。模型调用train() 【参数为mode,默认值为True】 会设置traing值等于mode值。调用eval() 【没有参数】实际执行会设置traing值为False,等同于train(False)。而最后 traing值会影响Dropou
2021-12-25 17:35:29
9271
1
原创 torch.cat函数深入详解
1、先看torch.cat 函数的官方解释:它的功能是将多个tensor类型矩阵的连接。它有两个参数,第一个是tensor元组或者tensor列表;第二个是dim,如果tensor是二维的,dim=0指在行上连接,dim=1指在列上连接。但是注意这里在行上连接,是扩展行进行连接,在列上连接是扩展列连接。2、下面看一些例子: a=torch.randn(2,3) print(a) b=torch.cat((a,a,a),1) print(b) '''#输出
2021-12-24 00:51:51
6347
原创 np.stack 函数最通俗本质理解
先看看官方对这个函数的功能解释:网址:https://numpy.org/doc/stable/reference/generated/numpy.stack.html?highlight=stack#numpy.stack有三个参数,第一个是放入多个用来堆叠列表,第二个参数是指定沿着某个轴(0,1,2…)去堆叠元素,默认值为0,第三个是可选参数输出到某个变量里。np.stack 要求堆叠的各个列表形状(shape属性)要相同。看的可能并不是很明白。这个函数的具体的堆叠细节官方有个没有解释。其实
2021-12-05 20:44:50
3471
1
原创 西农大数据实验6、线性回归
做这个实验按照文档步骤复制粘贴截图即可,但是会出现一些问题,下面是解决方案:问题1:hadoop fs -put xxx.xxx /user/root/ 上传文件报错hdfs没有启动,进入master 使用命令 start-dfs.sh问题2:导入numpy包出错File “/usr/local/bin/pip”, line 7, in from pip._internal.cli.main i解决安装numpy提示版本不适合,系统的环境python版本是python3.5使用下列命令即可解
2021-11-19 20:51:40
1984
原创 Mac sublime 设置 tensorflow代码自动补全,超简单!
自己找了好久如何设置tensorflow自动补全,还是没有找到合适的文章。后来自己摸索,弄出来了,有了自动补全的sublime更加强大了,写代码也变得更加舒服了!超简单教程第一步,如图所示,并点击package control第二步,点击install package第三步,输入tensorflow,点击安装,大功告成!...
2019-03-10 15:48:58
1327
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人