![](https://img-blog.csdnimg.cn/20190927151132530.png?x-oss-process=image/resize,m_fixed,h_224,w_224)
Pytorch
文章平均质量分 60
Pytorch应用
Raywit
这个作者很懒,什么都没留下…
展开
-
argparse模块中的参数action、dest使用
argparse模块中的参数action使用 在运行py文件时发现有些指令直接在py文件后面接上--path然后后面什么也没接,没接参数或者路径之类的,例如python demo.py --data_dir data --path,可以发现没有值赋给path,有人会说应该是缺省值default=xx之类,那既然有缺省值,我何必写上--path呢?这时就发现有这么一个参数 action=xx,写不写上 --pat原创 2021-01-06 11:15:53 · 11844 阅读 · 0 评论 -
PyTorch中torch.utils.data.DataLoader加载数据
torch.utils.data.DataLoader使用方法 一、参数设置 二、实际应用 DataLoader是PyTorch中的一种数据类型,在PyTorch架构中训练或者验证模型经常要使用它,那么怎么生成以及使用这样的数据类型? 一、参数设置torch.utils.data.DataLoader( dataset #数据加载 batch_size = 1 #批处理样本大小原创 2020-06-30 23:51:04 · 5087 阅读 · 2 评论 -
PyTorch保存网络结构以及参数【 torch.save()、torch.load() 】
对于pytorch保存网络参数,大家一般可以看到有 .pkl文件 以及 .pth文件,对于这两者有什么区别,以及如何保存网络参数等,本文就好好讲述一下。 一、保存方式 首先我们知道不论是保存模型还是参数都需要用到torch.save()。原创 2020-06-30 23:50:51 · 29735 阅读 · 4 评论 -
【PyTorch】常见错误: RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor)
【PyTorch】常见错误 错误:RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same 问题原因: 错误内容就在类型不匹配,根据报错内容可以看出Input type为原创 2020-06-30 23:50:35 · 72222 阅读 · 41 评论 -
【PyTorch】常见错误: RuntimeError:one of the variables needed for gradient computation has been modified
错误:RuntimeError: one of the variables needed for gradient computation has been modified by an inplace operation. 问题原因:pytorch版本导致的,0.3.0版本不会报错,0.4.0就报错;由于后者版本将Variable和Tensor合并为Tensor,但是inplace操作对Tensor不原创 2020-06-30 23:50:20 · 15642 阅读 · 4 评论 -
PyTorch的inplace的理解
inplace的理解 我们平时看到的 nn.ReLU(inplace=True)、nn.LeakyReLU(inplace=True),这些语句中的inplace是什么意思?inplace=True指的是进行原地操作,选择进行原地覆盖运算。 比如 x+=1则是对原值x进行操作,然后将得到的结果又直接覆盖该值。y=x+5,x=y则不是对x的原地操作。inplace=True操作的好处就是可以节省运算内存,不用多储存其他无原创 2020-06-30 23:50:04 · 9488 阅读 · 0 评论 -
nn.Sequential方法介绍
torch.nn.Sequential是一个Sequential容器,模块将按照构造函数中传递的顺序添加到模块中。另外,也可以传入一个有序模块。具体理解如下: ①普通构建网络:import torchimport torch.nn as nnclass Net(nn.Module): def __init__(self, n_feature, n_hidden, n_output)原创 2020-05-09 11:50:26 · 28804 阅读 · 2 评论 -
nn.conv2d方法介绍
1. conv2d用法格式:torch.nn.Conv2d(in_channels,out_channels,kernel_size,stride=1,padding=0,dilation=1,groups=1,bias=True)参数作用in_channels输入的通道数out_channels输出的通道数kennel_size卷积核大小(例如:kennel_size=2,大小则是2×2, 若kennel_size=(2,3),则是2×3)原创 2020-05-09 11:15:26 · 1674 阅读 · 0 评论 -
解决pycocotools安装问题
1. 方法: 对于安装pycocotools,总结起来有如下三种方法,希望挨个试下,指令都在cmd、终端或者虚拟环境下运行,不一样情形用不一样的方法。最简单的方法:pip install pycocotools (可能并不一定支持);查阅网上方法,发现pip install git+https://github.com/philferriere/cocoapi.git#subdirectory=Py原创 2020-05-08 22:52:43 · 3810 阅读 · 5 评论 -
pytorch中nn.Module方法
torch.nn是专门为神经网络设计的模块化接口,nn构建于autograd之上,可以用来定义和运行神经网络。 一、具体用法 nn.Module是nn中十分重要的类,包含网络各层的定义及forward方...原创 2020-05-07 12:05:42 · 746 阅读 · 1 评论 -
torch.norm方法介绍
可参考 官方文档torch.norm 一、具体用法功能:对输入的Tensor求范数;①torch.norm(input, p=2)→ float参数:– input (Tensor) – 输入张量– p (float,optional) – 范数计算中的幂指数值②torch.norm(input, p, dim, out=None)→ Tenso...原创 2020-05-07 10:51:44 · 2027 阅读 · 0 评论 -
pytorch中where方法介绍
一、具体用法 torch.where(condition, x, y)→ 其中 x,y 必须是Tensor类型※ 共有三个输入参数,第一个是condition判断条件,第二个是符合条件时取设置值x,第三个是不满足条件的设置值y; &nbs...原创 2020-05-07 09:17:36 · 7689 阅读 · 0 评论 -
argparse中add_argument方法介绍
add_argument介绍 add_argument:Python 内置的一个用于命令项选项与参数解析的模块,运行程序时,给定参数,通过调用给定的参数执行程序;1. 简单示例:※ 共分为三个步骤来进行:创建 ArgumentParser() 对象调用 add_argument() 方法添加参数使用 parse_args() 解析添加的参数import argpar...原创 2020-05-06 09:27:52 · 2402 阅读 · 0 评论