
pytorch
什么时候点菜
这个作者很懒,什么都没留下…
展开
-
pytorch张量
Tensor类的成员函数dim()可以返回张量的维度,shape属性与成员函数size()返回张量的具体维度分量,可以使用x.dim(),x.shape,x.size()原创 2021-11-05 14:57:01 · 89 阅读 · 0 评论 -
PyTorch学习日记(四)
书接上文,这次学到了使用pytorch基于经典网络架构训练图像分类模型一、数据预处理部分 1.1 数据增强和预处理:torchvision中tranforms模块自带功能import osimport matplotlib.pyplot as pltimport numpy as npimport torchfrom torch import nnimport torch.optim as optimimport torchvisionfrom torchvisio...原创 2021-09-30 22:10:55 · 1060 阅读 · 1 评论 -
PyTorch学习日记(四)
今天学习卷积神经网络的构建一、构建卷积神经网络处理mnist数据集 1.1 获取数据 分别构建训练集和测试集(验证集);用DataLoader来迭代取数据:import torchimport torch.nn as nnimport torch.optim as optimimport torch.nn.functional as Fimport matplotlib.pyplot as pltfrom torchvision import data...原创 2021-09-27 16:36:51 · 305 阅读 · 0 评论 -
PyTorch学习日记(三)
跟着唐宇迪学pytorch第三天,卷起来!卷积神经网络(CNN) 学了好几遍了,这里只记录一下该注意的点吧: 1、卷积核的通道数要和输入图片的通道数相同,并且每一个卷积核不同通道的参数值是不同的。例如,卷积核的三个通道对图片三个通道分别进行卷积计算后相加(之后加上bias偏置)得到一个特征值; 2、不同卷积层的卷积核大小可以不同,同一卷积层不同卷积核的大小需要相同,特征图的数量和卷积核的数量相同; 3、padding时填0的原因是,其...原创 2021-09-27 11:30:17 · 520 阅读 · 1 评论 -
PyTorch学习日记(二)
跟着唐宇迪pytorch学习的第二天,今天要用pytorch搭建神经网络进行气温预测。一、数据展示 1.1 读取数据: 使用pandas里的read_csv方法来读取csv数据文件并展示:import numpy as npimport pandas as pdimport matplotlib.pyplot as pltimport torchimport torch.optim as optimimport warningsimp...原创 2021-09-26 16:44:16 · 972 阅读 · 0 评论 -
PyTorch学习笔记(一)
记录一下跟着唐宇迪pytorch学习的学习过程。一、Pytorch简介与安装 我们知道,GPU的并行计算能力高, Pytorch做的就是将矩阵计算传送到GPU里进行计算。在2019年以前Tensorflow是占主流的,而2019年之后Tensorflow使用人数大量下降,而pytorch迅速上升,原因是顶会和许多期刊使用pytorch的越来越多。 安装:pip install torch,以及torchvision,可选择CPU版本和GPU版本,由于实验室里的电脑没显卡...原创 2021-09-24 22:20:45 · 947 阅读 · 1 评论