码迷,mamicode.com
首页 >  
搜索关键字:pytorch    ( 1103个结果
PyToune:一款类Keras的PyTorch框架
官方文档:https://pytoune.org/index.html ...
分类:其他好文   时间:2018-08-04 11:56:55    阅读次数:125
pytorch learning rate decay
关于learning rate decay的问题,pytorch 0.2以上的版本已经提供了torch.optim.lr_scheduler的一些函数来解决这个问题。 我在迭代的时候使用的是下面的方法。 classtorch.optim.lr_scheduler.MultiStepLR(optimi ...
分类:其他好文   时间:2018-08-04 11:38:25    阅读次数:258
pytorch中的cat、stack、tranpose、permute、unsqeeze
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
Pytorch中的Batch Normalization操作
之前一直和小伙伴探讨batch normalization层的实现机理,作用在这里不谈,知乎上有一篇paper在讲这个,链接 这里只探究其具体运算过程,我们假设在网络中间经过某些卷积操作之后的输出的feature map的尺寸为4×3×2×2 4为batch的大小,3为channel的数目,2×2为 ...
分类:其他好文   时间:2018-07-18 23:24:33    阅读次数:17915
pytorch保存模型等相关参数,利用torch.save(),以及读取保存之后的文件
本文分为两部分,第一部分讲如何保存模型参数,优化器参数等等,第二部分则讲如何读取。 假设网络为model = Net(), optimizer = optim.Adam(model.parameters(), lr=args.lr), 假设在某个epoch,我们要保存模型参数,优化器参数以及epoc ...
分类:其他好文   时间:2018-07-15 00:50:52    阅读次数:486
pytorch将cpu训练好的模型参数load到gpu上,或者gpu->cpu上
假设我们只保存了模型的参数(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
在Windows上安装pytorch
电脑环境为 :windows10 python3.5 cuda8.0 可以在官网查询各个版本的安装方式 这里使用pip的安装方式,通过以下命令: 在安装完成后在python中验证安装: 可能出现的问题: 在import torch时出现 from torch._C import * ImportEr ...
分类:Windows程序   时间:2018-07-09 20:10:47    阅读次数:2859
PyTorch 使用心得
PyTorch 使用心得 模板 ...
分类:其他好文   时间:2018-07-04 23:39:04    阅读次数:172
Lstm Cell in detail and how to implement it by pytorch
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
1103条   上一页 1 ... 93 94 95 96 97 ... 111 下一页
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!