![](https://img-blog.csdnimg.cn/2019092715111047.png?x-oss-process=image/resize,m_fixed,h_224,w_224)
Pytorch
zxfhahaha
这个作者很懒,什么都没留下…
展开
-
【pytorch】model.eval()和model.train()
eval()self.eval()和self.train(False)等价eval()在测试之前加,否则有输入数据即使不训练,它也会改变权值pytorch会自己把BatchNormalization和DropOut固定住,不会取平均,而是用训练好的值。train()model.train()让model变成训练模式...原创 2020-07-01 19:46:19 · 4060 阅读 · 0 评论 -
【pytorch】Loss functions 损失函数总结
首先我们来看一下nn和nn.functional有什么区别nn.functional.xxx是函数接口,而nn.Xxx是nn.functional.xxx的类封装,并且nn.Xxx都继承于一个共同祖先nn.Module。这一点导致nn.Xxx除了具有nn.functional.xxx功能之外,内部附带了nn.Module相关的属性和方法,例如train(), eval(),load_state_...原创 2019-10-15 18:16:48 · 1999 阅读 · 0 评论 -
【pytorch】tensor的操作
转置 transpose和permute对二维矩阵来说,转置就是把矩阵的行列互换 也就是原来是X[i][j]X[i][j]X[i][j]转置后变成X[j][i]X[j][i]X[j][i]torch.transposetranspose函数介绍torch.transpose(input, dim0, dim1) → Tensortranspose就是把input 的第dim0维和di...原创 2019-10-15 10:16:04 · 1250 阅读 · 0 评论 -
【pytorch】批训练 DataLoader
我们训练模型要进行批训练的时候,就涉及到每一批选取什么数据的问题,pytorch的DataLoader就帮我们包装了数据,它能帮我们有效迭代数据,这样就可以进行批训练。pytorch 数据加载到模型的流程pytorch 的数据加载到模型的操作顺序是这样的:① 创建一个 Dataset 对象② 创建一个 DataLoader 对象③ 循环这个 DataLoader 对象,将img, lab...原创 2019-10-14 15:53:52 · 2489 阅读 · 0 评论