pytorch
文章平均质量分 88
baidu_huihui
这个作者很懒,什么都没留下…
展开
-
对pytorch中x = x.view(x.size(0), -1) 的理解说明
这篇文章主要介绍了对pytorch中x = x.view(x.size(0), -1) 的理解说明,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧在pytorch的CNN代码中经常会看到x.view(x.size(0), -1)首先,在pytorch中的view()函数就是用来改变tensor的形状的,例如将2行3列的tensor变为1行6列,其中-1表示会自适应的调整剩余的维度a = torch.Tensor(2,3)print(a)# tensor([[0....原创 2021-06-22 14:01:30 · 1843 阅读 · 0 评论 -
pytorch: 准备、训练和测试自己的图片数据
大部分的pytorch入门教程,都是使用torchvision里面的数据进行训练和测试。如果我们是自己的图片数据,又该怎么做呢?一、我的数据我在学习的时候,使用的是fashion-mnist。这个数据比较小,我的电脑没有GPU,还能吃得消。关于fashion-mnist数据,可以百度,也可以点此了解一下,数据就像这个样子:下载地址:https://github.com/zalandoresearch/fashion-mnist但是下载下来是一种二进制文件,并不是图片,...原创 2020-10-21 13:31:04 · 861 阅读 · 1 评论 -
Windows系统安装Miniconda搭建深度学习环境
下载MinicondaMiniconda的官网:https://docs.conda.io/en/latest/miniconda.html官网下载速度慢,直接去清华镜像网站下载红框,框出来的版本:https://mirrors.tuna.tsinghua.edu.cn/anaconda/miniconda/安装Miniconda下载完成,我们开始安装:一路Next,虚拟环境操作添加国内源快捷键:win + r,打开命令行,如图:出现如下图:执行:conda confi...原创 2020-09-22 19:58:16 · 1511 阅读 · 0 评论 -
x.view(x.size(0), -1)的解释
pytorch教程中,用cnn实现mnist的分类。其中class中有一段代码:def forward(self, x): x = self.conv1(x) x = self.conv2(x) # t = x.size(0) 0->50,1->32,2->7,3->7 x = x.view(x.size(0), -1) # flatten the output of conv2 to (batch_size, 32 ...转载 2020-08-20 05:06:15 · 932 阅读 · 0 评论 -
CUDA+cuDNN+GPU版pytorch安装
GPU版pytorch安装设备环境:Windows10 64bitGPU: GeForce GTX 1060 TiPython3.7 64bit1 下载安装CUDA以及cuDNN1.1 查看电脑显卡对应的CUDA驱动版本1.2 下载安装CUDA下载地址:https://developer.nvidia.com/cuda-downloads?target_os=Windows&target_arch=x86_64&target_version=10&..原创 2020-08-06 20:09:32 · 1080 阅读 · 0 评论 -
TensorRT安装及使用教程
分类专栏:人工智能TensorFlow1 什么是TensorRT一般的深度学习项目,训练时为了加快速度,会使用多GPU分布式训练。但在部署推理时,为了降低成本,往往使用单个GPU机器甚至嵌入式平台(比如 NVIDIA Jetson)进行部署,部署端也要有与训练时相同的深度学习环境,如caffe,TensorFlow等。由于训练的网络模型可能会很大(比如,inception,resnet...原创 2019-11-14 22:45:40 · 2162 阅读 · 1 评论 -
莫烦PyTorch学习笔记(二)——回归
莫烦PyTorch学习笔记(二)——回归本文主要是用PyTorch来实现一个简单的回归任务。编辑器:spyder1.引入相应的包及生成伪数据import torchimport torch.nn.functional as F # 主要实现激活函数import matplotlib.pyplot as plt # 绘图的工具from torch.autogra...转载 2019-05-26 22:15:34 · 485 阅读 · 1 评论 -
莫烦PyTorch学习笔记(一)——Torch或Numpy
1.什么是NumpyNumpy系统是Python的一种开源的数值计算扩展,用python实现的科学计算包。这种工具可用来存储和处理大型矩阵,包括强大的N维数组对象Array,比较成熟的函数库等。numpy和稀疏矩阵运算包scipy配合使用更加方便。2.用Numpy还是TorchTorch自称为神经网络界的Numpy,它能将torch产生的tensor放在GPU中加速运算,就想...转载 2019-05-26 22:10:05 · 233 阅读 · 0 评论 -
Windows下Pytorch的安装及运行
Windows下Pytorch的安装及运行所属专栏:深度学习Pytorch的安装分为3个部分:Anaconda的安装、cuda安装、pytorch安装Anaconda建议使用最新版本,使用旧版本的,需要升级python等库,会遇到问题。解决问题的时间大于直接重新安装anaconda的时间,所以选择更新anacondaAnaconda下载地址为:https://www.an...转载 2019-04-14 10:22:18 · 10051 阅读 · 0 评论