![](https://img-blog.csdnimg.cn/20201014180756926.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
gie+
这个作者很懒,什么都没留下…
展开
-
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 · 8032 阅读 · 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 · 6246 阅读 · 0 评论