pytorch学习笔记
文章平均质量分 73
介绍pytorch的使用
xz1308579340
这个作者很懒,什么都没留下…
展开
-
torch tensor操作
viewview()用于改变tensor的形状meshgridhttps://blog.csdn.net/weixin_39504171/article/details/106356977torch.round()四舍五入取整gather四维gatherhttps://stackoverflow.com/questions/53471716/index-pytorch-4d-tensor-by-values-in-2d-tensorhttps://zhuanlan.zhihu.com/p/原创 2022-05-31 18:49:39 · 623 阅读 · 2 评论 -
torch的maximum与max以及导出onnx
文章目录1、torch.max()2、torch.min()3、 torch.maximum()4、torch.minimum()获得 Tensor 元素中的最大值、最小值1、torch.max()torch.max(input, dim, keepdim=False)返回命名元组(最大值,最大值索引),最大值是给定维度中的最大值,索引为在对应维度中的索引当有多个最大值时,返回第一个最大值的索引当没有指定维度 dim 时,则返回所有元素中的最大值(single value)input:输入张量原创 2022-05-22 19:21:00 · 1831 阅读 · 0 评论 -
将torch模型转化为onnx模型
函数torch.onnx.export(model, args, f, export_params=True, verbose=False, training=False, input_names=None, output_names=None, aten=False, export_raw_ir=False, operator_export_type=None,opset_version=None, _retain_param_name=True, do_constant_f原创 2022-05-22 12:07:53 · 4083 阅读 · 0 评论 -
DCGAN-极简入门教程
网上GAN的教程太多了,这边也谈一下自己的理解首先GAN由两部分组成1.图片分类网络,这是一个二分类网络,可以是alxnet ,vgg,resnet任何一个,复制对图片进行二分类,是真实图片还是生成的图片2.图片生成网络,输入是一个随机噪声,输出是一张图片,使用的是反卷积层相信学过深度学习的都能写出这两个网络,当然如果你写不出来,没关系,有人替你写好了首先是图片分类网络:简单来说就是c...原创 2020-05-02 01:38:48 · 729 阅读 · 0 评论 -
pytorch学习笔记-mnist
pytorch学习笔记1-mnist摘要:实现mnist手写数字识别,test准确率98% 1.构建网络 nn.Moudle是pytorch官方指定的编写Net模块,在init函数中添加需要使用的层,在foeword中定义网络流向。下面详细解释各层: conv1层:输入channel = 1 ,输出chanael = 10,滤波器5*5 maxpooling = 2*2 ...原创 2018-03-11 22:51:44 · 3978 阅读 · 1 评论 -
pytorch学习笔记1: Tensor
pytorch学习笔记1: Tensor参考网址pytorch学习笔记1: Tensor1、Tensors建立5*3的矩阵,未初始化建立随机初始化矩阵建立零初始化矩阵,数据类型是Long建立一个tensor数据来源于data在原有tnesor的基础上形成新的tensor,会继承原有tensor的shapee和dtype等属性,当然我么也可以修改这些属性获取tensor...原创 2018-04-25 13:52:01 · 2024 阅读 · 0 评论 -
pytorch学习笔记3:网络的构建
pytorch学习笔记3:网络的构建参考网址pytorch学习笔记3:网络的构建构建网络定义一个网络loss FunctionBackprop更新权值构建网络我们可以通过torch.nn包来构建网络,现在你已经看过了autograd,nn在autograd的基础上定义模型和求微分。一个nn.Module包括很多层,forward方法返回output。...原创 2018-04-25 21:38:34 · 513 阅读 · 0 评论 -
pytorch Vgg源码阅读
要实现VGG16,主要需要两个类VGGVGG16VGG16调用VGG这两个类的实现在VGG.PY中实现首先我们来看看VGG类的实现class VGG(nn.Module): def __init__(self, features, num_classes=1000, init_weights=True): super(VGG, self).__init__(...原创 2019-03-16 22:27:29 · 1380 阅读 · 0 评论 -
使用pytorch可视化中间层的结果
一直比较想知道图片经过卷积之后中间层的结果,于是使用pytorch写了一个脚本查看,先看效果这是原图我们使用的VGG16,包含RULE层总共有30层可以可视化的结果,原图大小为224224,经过第一层后大小为64224*224,下面是可视化的结果:...原创 2019-01-02 16:38:15 · 16837 阅读 · 10 评论 -
Pytorch的内置模型
Pytorch的内置模型翻译自https://pytorch.org/docs/stable/torchvision/models.htmlTORCHVISION.MODELS文章目录Pytorch的内置模型torchvision.models中包含了如下模型你可以通过随机权重来初始化一个模型我们给他们提供了预训练的模型,使用torch.utils.model_zoo ,通过让参数pr...原创 2018-12-29 11:15:10 · 12091 阅读 · 0 评论 -
pytorch学习笔记4:网络和损失函数的可视化
1.效果2.环境1.pytorch2.visdom3.python3.53.用到的代码# coding:utf8import torchfrom torch import nn, optim # nn 神经网络模块 optim优化函数模块from torch.utils.data import DataLoaderfrom torch.autograd import Va...原创 2018-12-15 16:27:04 · 6786 阅读 · 0 评论 -
pytorch源码解读之resnet.py的实现
文章目录调用pytorch内置的模型的方法解读模型源码Resnet.py包含的库文件该库定义了6种Resnet的网络结构,包括每种网络都有训练好的可以直接用的.pth参数文件Resnet中大多使用3*3的卷积定义如下如何定义不同大小的Resnet网络Resnet18resnet34Resnet类我们来看看网络的forward过程layer层的实现在理解Resnet之前,我们先要理解一下Blocki...原创 2019-01-17 17:13:48 · 5381 阅读 · 1 评论