码迷,mamicode.com
首页 >  
搜索关键字:pytorch    ( 1103个结果
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
[Pytorch] 3D卷积尺寸计算
公式来自官方文档,戳->(Conv3d — PyTorch master documentation) 本文仅作记录,顺便练习Latex语法 \(D_{out}=\frac{D_{in}+2\times padding[0]-dilation[0]\times(kernel\_size[0]-1)- ...
分类:其他好文   时间:2020-05-31 16:08:36    阅读次数:230
pytorch创建tensor数据
import torch 一、传入数据 tensor只能传入数据 可以传入现有的数据列表或矩阵 # 当是标量时候,即只有一个数据时候,[]括号是可以省略的 torch.tensor(2) tensor(2) # 如果是向量或矩阵,必须有[]括号 torch.tensor([2, 3]) tensor ...
分类:其他好文   时间:2020-05-30 01:21:23    阅读次数:73
1103条   上一页 1 ... 22 23 24 25 26 ... 111 下一页
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!