码迷,mamicode.com
首页 >  
搜索关键字:caffe2 PyTorch    ( 1125个结果
pytorch 中的view、reshape、permute、transpose
pytorch 中的view、reshape、permute、transpose? view相当于numpy中resize()的功能,但是用法可能不太一样.可以参考:https://blog.csdn.net/york1996/article/details/81949843 view只能用在con ...
分类:其他好文   时间:2020-06-05 21:07:28    阅读次数:61
pytorch教程之nn.Sequential类详解——使用Sequential类来自定义顺序连接模型
前言:类似于keras中的序贯模型,当一个模型较简单的时候,我们可以使用torch.nn.Sequential类来实现简单的顺序连接模型。这个模型也是继承自Module类的,关于这个类,后面的文章会详细介绍。 一、关于Sequential类的简介先来看一下它的定义吧,在之前,我们首先需要明确一个特别 ...
分类:其他好文   时间:2020-06-05 16:49:46    阅读次数:181
PyTorch自定义OP导出ONNX
根据PyTorch的官方文档,需要用Function封装一下,为了能够导出ONNX需要加一个symbolic静态方法: class relu5_func(Function): @staticmethod def forward(ctx, input): return relu5_cuda.relu5 ...
分类:其他好文   时间:2020-06-04 14:12:08    阅读次数:193
pytorch安装
如果你使用 conda install pytorch=1.3.0 torchvision cudatoolkit=10.0 -c pytorch 来安装pytorch,并且添加了清华镜像源,但还是由于网络原因下载失败,你只需要把-c pytorch去掉: conda install pytorch ...
分类:其他好文   时间:2020-06-04 01:28:31    阅读次数:108
pytorch基本运算:加减乘除、对数幂次等
1、加减乘除 a + b = torch.add(a, b) a - b = torch.sub(a, b) a * b = torch.mul(a, b) a / b = torch.div(a, b) import torch a = torch.rand(3, 4) b = torch.ran ...
分类:其他好文   时间:2020-06-03 23:40:49    阅读次数:175
pytorch 数据维度变换
view、reshape 两者功能一样:将数据依次展开后,再变形 变形后的数据量与变形前数据量必须相等。即满足维度:ab...f = xy*...*z reshape是pytorch根据numpy中的reshape来的 import torch a = torch.rand(2, 3, 2, 3) ...
分类:其他好文   时间:2020-06-03 00:26:51    阅读次数:46
pycharm远程调试服务器代码
主要目的: 代码上传到服务器上,环境也搭在了服务器上,本地pycharm能够随时调试和调整远端代码 #配置远端解释器 1. conda info -e 2. conda acivate pytorch_gpu 这是我的环境名 3. 查看当前环境python解释器位置 >>> python >>> i ...
分类:其他好文   时间:2020-06-02 18:59:19    阅读次数:78
Pytorch之conv2d
cross-correlation(互相关、交叉相关): Coutj 第j个输出Channel(或由第j个Filter输出) 对于每个Coutj (或每个Filter)和Ni个Kernal构成的滑动窗口来说: 输出点(neural)(为Kernal滑动位置和Filter的函数)s值为: 该Filte ...
分类:其他好文   时间:2020-06-02 00:14:44    阅读次数:157
使用 Google Colab上的PyTorch YOLOv3
作者|Hervind Philipe 编译|VK 来源|Towards Data Science 对于计算机视觉爱好者来说,YOLO (You Only Look Once)是一个非常流行的实时对象检测概念,因为它非常快,而且性能非常好。 在本文中,我将共享处理视频的代码,以获取谷歌Colab内每个 ...
分类:其他好文   时间:2020-05-31 19:47:35    阅读次数:91
NVIDIA深度学习Tensor Core性能解析第二季
虽然RNN可能会有加速,但DeepBench和NVIDIA目前仅支持单精度RNN推理。 NVIDIA Caffe2测试之ResNet50和ImageNet 虽然内核和深度学习数学运算可能很有用,但实际应用中是使用真实数据集进行训练的。使用标准的ILSVRC 2012图片集,在ImageNet上通过R ...
分类:其他好文   时间:2020-05-31 16:20:58    阅读次数:68
1125条   上一页 1 ... 22 23 24 25 26 ... 113 下一页
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!