码迷,mamicode.com
首页 >  
搜索关键字:pytorch    ( 1103个结果
PyTorch 60 分钟入门教程:数据并行处理
可选择:数据并行处理(文末有完整代码下载) 作者:Sung Kim?和?Jenny Kang 在这个教程中,我们将学习如何用 DataParallel 来使用多 GPU。 通过 PyTorch 使用多个 GPU 非常简单。你可以将模型放在一个 GPU: device = torch.device(" ...
分类:其他好文   时间:2018-12-11 13:01:08    阅读次数:173
Pytorch-nn.ConvTransposed2d()
ConvTransposed2d()其实是Conv2d()的逆过程,其参数是一样的 Conv2d(): output = (input+2*Padding-kernelSize) / stride + 1(暂时不考虑outputPadding 注意:outputPadding只是在一边Padding ...
分类:其他好文   时间:2018-12-10 14:17:52    阅读次数:449
pytorch:EDSR 生成训练数据的方法
Pytorch:EDSR 生成训练数据的方法 引言 Winter is coming 正文 提供的 是用来包装你的数据的工具. 所以你要将自己的 (numpy array 或其他) 数据形式装换成 , 然后再放进这个包装器中. 使用 有什么好处呢? 就是他们帮你 有效地迭代数据 , 举例: x = ...
分类:其他好文   时间:2018-12-09 14:09:37    阅读次数:517
[PyTorch]PyTorch中模型的参数初始化的几种方法(转)
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~本文目录1. xavier初始化2. kaiming初始化3. 实际使用中看到的初始化3.1 ResNeXt,densenet中初始化3.2 wide residual networks中初始化(MSRinit)~~~~~~ ...
分类:其他好文   时间:2018-12-08 17:55:00    阅读次数:1963
pytorch faster_rcnn
代码地址:https://github.com/jwyang/faster-rcnn.pytorch fasterRCNN.train():这个不是让网络进行训练,而是让module in training mode,有些module在traing model和testing model下不同 即s ...
分类:其他好文   时间:2018-12-07 21:20:14    阅读次数:187
[Pytorch]Pytorch 保存模型与加载模型(转)
转自: "知乎" 目录:保存模型与加载模型冻结一部分参数,训练另一部分参数采用不同的学习率进行训练1.保存模型与加载简单的保存与加载方法: 保存整个网络 torch.save(net, PATH) 保存网络中的参数, 速度快,占空间少 torch.save(net.state_dict(),PATH ...
分类:其他好文   时间:2018-12-06 00:05:03    阅读次数:204
pytorch安装(使用pip3装到conda环境下)
之前在windows上使用pip3的方式安装过pytorch,虽然偶尔也会HTTP Error,安装失败,但是基本还是可以安装成功的。上 这一次在ubuntu上装了anaconda,想着之后的库都用conda安装,而不使用ubuntu自带的python2.7和python3.5,因此使用如下命令安装 ...
分类:其他好文   时间:2018-12-02 22:41:38    阅读次数:604
【OCR技术系列之六】文本检测CTPN的代码实现
这几天一直在用Pytorch来复现文本检测领域的CTPN论文,本文章将从数据处理、训练标签生成、神经网络搭建、损失函数设计、训练主过程编写等这几个方面来一步一步复现CTPN。CTPN算法理论可以参考 "这里" 。 训练数据处理 我们的训练选择天池ICPR2018和MSRA_TD500两个数据集,天池 ...
分类:其他好文   时间:2018-12-02 18:34:22    阅读次数:3417
pytorch代码中同时包含训练和测试代码时显存爆炸
原因在于没有使用torch.no_grad()函数。在查看验证集和测试集表现时,应使用类似这样的代码 ...
分类:其他好文   时间:2018-12-02 13:34:27    阅读次数:591
Cloud Native Weekly | KubeCon首登中国,华为云亮相KubeCon
KubeCon首登中国,Kubernetes将如何再演进?华为云亮相KubeCon 2018,开源两款技术全新上线微软云服务又罢工了。Facebook 发布 PyTorch 1.0 预览版本,集成 Google,AWS 和 Microsoft Azure 云服务
分类:其他好文   时间:2018-11-26 17:23:36    阅读次数:207
1103条   上一页 1 ... 85 86 87 88 89 ... 111 下一页
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!