pytorch
文章平均质量分 81
pytorch日常笔记
Teeyohuang
数据挖掘工程师(菜鸡)
展开
-
离线安装pytorch
离线安装pytorch核心思想:搭梯子翻出去 用网址链接直接用浏览器下载安装包,然后拷贝回linux系统中本地离线安装。首先报一下我自己的参数:Anaconda3,python3.7,CUDA10.1.243,CUDNN7.6.3然后我就在官网上找到适合我的,写这篇博文的时候是1.4版本为stable版本,结合自己的参数选择如下界面:官网的安装指令为:conda i...原创 2020-05-01 16:20:21 · 6002 阅读 · 0 评论 -
torch.gather() 和torch.sactter_()的用法简析
torch.gather(input, dim, index, out=None) 和 torch.scatter_(dim, index, src)是一对作用相反的方法 先来看torch.gather, 核心操作其实就是这样:out[i][j][k] = input[index[i][j][k]] [j][k] # if dim == 0out[i][j][k] =...原创 2018-08-29 17:02:40 · 16556 阅读 · 6 评论 -
Pytorch打怪路(三)Pytorch创建自己的数据集2
前面一篇写创建数据集的博文--- Pytorch创建自己的数据集1 是介绍的应用于图像分类任务的数据集,即输入为一个图像和它的类别数字标签,本篇介绍输入的标签label亦为图像的数据集,并包含一些常用的处理手段。比如做图像语义分割时就会用到这种数据输入方式。1、数据集简介以VOC2012数据集为例,图像是RGB3通道的,label是1通道的,(其实label原来是几通...原创 2018-08-27 11:01:22 · 18962 阅读 · 15 评论 -
Pytorch出现 raise NotImplementedError
作者:teeyohuang邮箱:teeyohuang@163.com本文系原创,供交流学习使用,转载请注明出处,谢谢1.问题背景:用pytorch跑网络的时候,一开始就出现这个问题2.问题所在经过反复检查,发现是 forward函数 出了问题,没检测到forward函数但我明明写了 def forward ....的,后来发现这一段代码多缩进了一个Tap位……一定要仔细啊……注意观察和上面的 de...原创 2018-07-15 19:07:08 · 24234 阅读 · 37 评论 -
Pytorch打怪路(一)pytorch进行CIFAR-10分类(5)测试
pytorch进行CIFAR-10分类(5)测试我的系列博文:Pytorch打怪路(一)pytorch进行CIFAR-10分类(1)CIFAR-10数据加载和处理Pytorch打怪路(一)pytorch进行CIFAR-10分类(2)定义卷积神经网络Pytorch打怪路(一)pytorch进行CIFAR-10分类(3)定义损失函数和优化器Pytorch打怪路(一)pyt...原创 2018-02-01 13:24:38 · 17833 阅读 · 33 评论 -
torch.stack()的使用
废话不多说直接上图如图有a, b, c 三个 3x3的Tensor, 如果我想把这三个tensor的最后一个维度的元素相叠加,形成一个新的tensor输入 d=torch.stack( (a,b,c) ,dim = 2)就会发现他们在第二个维度像叠加,具体来说变化如下图所示d[0][0]的位置是由a[0][0] 的 [1] , b[0][0] 的 [10], c[0][0] 的 [100]...原创 2018-05-18 12:56:30 · 121638 阅读 · 13 评论 -
Pytorch打怪路(三)Pytorch创建自己的数据集1
之前讲的例子,程序都是调用的datasets方法,下载的torchvision本身就提供的数据,那么如果想导入自己的数据应该怎么办呢?本篇就讲解一下如何创建自己的数据集。还有第二篇……Pytorch打怪路(三)Pytorch创建自己的数据集21.用于分类的数据集以mnist数据集为例这里的mnist数据集并不是torchvision里面的,而是我自己的以图片格式保存的数据集...原创 2018-03-16 21:54:21 · 48218 阅读 · 39 评论 -
Pytorch打怪路(二)pytorch进行mnist训练和测试
1.简述程序来自pytorch官方tutorial给的examples,链接为:https://github.com/pytorch/examples/blob/master/mnist/main.py我主要进行一下注释工作和实验程序2.代码from __future__ import print_function #这个是python当中让print都以python3的原创 2018-02-02 21:18:04 · 29865 阅读 · 3 评论 -
Linux系统(Ubantu16.04)安装Pytorch
1.首先安装Anaconda安装教程参考我的博文:Linux系统安装Anaconda2.进入pytorch官网选择自己对应的版本选择conda方式安装,python版本是3.6,CUDA版本是9这是我电脑的配置,如何查看你自己的配置?terminal窗口输入python如何查看CUDA版本?右转我的博客:如何查看CUDA版本和原创 2018-01-17 10:29:42 · 9852 阅读 · 0 评论 -
Pytorch打怪路(一)pytorch进行CIFAR-10分类(4)训练
pytorch进行CIFAR-10分类(4)训练我的系列博文:Pytorch打怪路(一)pytorch进行CIFAR-10分类(1)CIFAR-10数据加载和处理Pytorch打怪路(一)pytorch进行CIFAR-10分类(2)定义卷积神经网络Pytorch打怪路(一)pytorch进行CIFAR-10分类(3)定义损失函数和优化器1、简述原创 2018-01-31 22:04:26 · 19412 阅读 · 18 评论 -
Pytorch打怪路(一)pytorch进行CIFAR-10分类(3)定义损失函数和优化器
pytorch进行CIFAR-10分类(3)定义损失函数和优化器我本打算把这一步的内容也归并到第二步定义网络模型中去,因为我觉得它们其实可以宏观上看成一个大部分,但是既然官方教程中分成了5步,那我也就依照这个来,免得各位看官被我给搞昏了。Pytorch打怪路(一)pytorch进行CIFAR-10分类(1)CIFAR-10数据加载和处理Pytorch打怪路(一)pyto原创 2018-01-31 16:14:20 · 14566 阅读 · 3 评论 -
Pytorch打怪路(一)pytorch进行CIFAR-10分类(2)定义卷积神经网络
Pytorch打怪路(一)pytorch进行CIFAR-10分类(2)定义卷积神经网络注:官方文档地址-- http://pytorch.org/docs/0.3.0/index.html我的系列博文Pytorch打怪路(一)pytorch进行CIFAR-10分类(1)CIFAR-10数据加载和处理Pytorch打怪路(一)pytorch进行CIFAR-10分类(2)原创 2018-01-31 00:41:36 · 19195 阅读 · 15 评论 -
Pytorch打怪路(一)pytorch进行CIFAR-10分类(1)CIFAR-10数据加载和处理
pytorch进行CIFAR-10分类(1)CIFAR-10数据加载和处理1、写在前面的话这一篇博文的内容主要来自于pytorch的官方tutorial,然后根据自己的理解把cifar10这个示例讲一遍,权当自己做笔记。因为这个cifar10是官方example,所以适合我们拿来先练手,至少能保证代码的正确性。之所以第一篇pytorch的博文(其实之前还写了篇如何安装pytorch)...原创 2018-01-30 21:26:58 · 65098 阅读 · 34 评论