官方文档:https://pytoune.org/index.html ...
分类:
其他好文 时间:
2018-08-04 11:56:55
阅读次数:
125
关于learning rate decay的问题,pytorch 0.2以上的版本已经提供了torch.optim.lr_scheduler的一些函数来解决这个问题。 我在迭代的时候使用的是下面的方法。 classtorch.optim.lr_scheduler.MultiStepLR(optimi ...
分类:
其他好文 时间:
2018-08-04 11:38:25
阅读次数:
258
Cat对数据沿着某一维度进行拼接。cat后数据的总维数不变.比如下面代码对两个2维tensor(分别为2*3,1*3)进行拼接,拼接完后变为3*3还是2维的tensor。import torch torch.manual_seed(1) x = torch.randn(2,3) y = torch.... ...
分类:
其他好文 时间:
2018-08-01 01:02:07
阅读次数:
407
之前一直和小伙伴探讨batch normalization层的实现机理,作用在这里不谈,知乎上有一篇paper在讲这个,链接 这里只探究其具体运算过程,我们假设在网络中间经过某些卷积操作之后的输出的feature map的尺寸为4×3×2×2 4为batch的大小,3为channel的数目,2×2为 ...
分类:
其他好文 时间:
2018-07-18 23:24:33
阅读次数:
17915
本文分为两部分,第一部分讲如何保存模型参数,优化器参数等等,第二部分则讲如何读取。 假设网络为model = Net(), optimizer = optim.Adam(model.parameters(), lr=args.lr), 假设在某个epoch,我们要保存模型参数,优化器参数以及epoc ...
分类:
其他好文 时间:
2018-07-15 00:50:52
阅读次数:
486
假设我们只保存了模型的参数(model.state_dict())到文件名为modelparameters.pth, model = Net() 1. cpu -> cpu或者gpu -> gpu: checkpoint = torch.load('modelparameters.pth') mod ...
分类:
其他好文 时间:
2018-07-15 00:16:29
阅读次数:
215
1.本科及以上学历,2年以上图像算法开发的经验; 2.熟练掌握C++,熟练掌握Python并行开发,界面开发; 3.熟悉SVM、CNN、SSD、YOLOv2灯机器学习模型,掌握数字图像处理的基础 4.熟悉至少一种主流深度学习算法框架(如Caffe,Caffe2,Mxnet,PyTorch,Tenso ...
分类:
编程语言 时间:
2018-07-11 22:53:28
阅读次数:
204
电脑环境为 :windows10 python3.5 cuda8.0 可以在官网查询各个版本的安装方式 这里使用pip的安装方式,通过以下命令: 在安装完成后在python中验证安装: 可能出现的问题: 在import torch时出现 from torch._C import * ImportEr ...
分类:
Windows程序 时间:
2018-07-09 20:10:47
阅读次数:
2859
Refer to : https://medium.com/@andre.holzner/lstm-cells-in-pytorch-fab924a78b1c http://colah.github.io/posts/2015-08-Understanding-LSTMs/ LSTM cells i ...
分类:
其他好文 时间:
2018-07-01 10:27:37
阅读次数:
331