前言:类似于keras中的序贯模型,当一个模型较简单的时候,我们可以使用torch.nn.Sequential类来实现简单的顺序连接模型。这个模型也是继承自Module类的,关于这个类,后面的文章会详细介绍。 一、关于Sequential类的简介先来看一下它的定义吧,在之前,我们首先需要明确一个特别 ...
分类:
其他好文 时间:
2020-06-05 16:49:46
阅读次数:
181
根据PyTorch的官方文档,需要用Function封装一下,为了能够导出ONNX需要加一个symbolic静态方法: class relu5_func(Function): @staticmethod def forward(ctx, input): return relu5_cuda.relu5 ...
分类:
其他好文 时间:
2020-06-04 14:12:08
阅读次数:
193
如果你使用 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
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
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能够随时调试和调整远端代码 #配置远端解释器 1. conda info -e 2. conda acivate pytorch_gpu 这是我的环境名 3. 查看当前环境python解释器位置 >>> python >>> i ...
分类:
其他好文 时间:
2020-06-02 18:59:19
阅读次数:
78
cross-correlation(互相关、交叉相关): Coutj 第j个输出Channel(或由第j个Filter输出) 对于每个Coutj (或每个Filter)和Ni个Kernal构成的滑动窗口来说: 输出点(neural)(为Kernal滑动位置和Filter的函数)s值为: 该Filte ...
分类:
其他好文 时间:
2020-06-02 00:14:44
阅读次数:
157
作者|Hervind Philipe 编译|VK 来源|Towards Data Science 对于计算机视觉爱好者来说,YOLO (You Only Look Once)是一个非常流行的实时对象检测概念,因为它非常快,而且性能非常好。 在本文中,我将共享处理视频的代码,以获取谷歌Colab内每个 ...
分类:
其他好文 时间:
2020-05-31 19:47:35
阅读次数:
91
公式来自官方文档,戳->(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
import torch 一、传入数据 tensor只能传入数据 可以传入现有的数据列表或矩阵 # 当是标量时候,即只有一个数据时候,[]括号是可以省略的 torch.tensor(2) tensor(2) # 如果是向量或矩阵,必须有[]括号 torch.tensor([2, 3]) tensor ...
分类:
其他好文 时间:
2020-05-30 01:21:23
阅读次数:
73