发布于2019-08-07 14:54 阅读(1158) 评论(0) 点赞(2) 收藏(3)
英文手册 https://pytorch.org/docs/stable/index.html
中文手册推荐使用 https://pytorch-cn.readthedocs.io/zh/latest/
搬砖的学习
PyTorch常见小问题
H x W x C
,Pytorch的Tensor为C X H X W
(搞笑问题) num_workers
:使用多进程加载的进程数,0代表不使用多进程PyTorch数据集
Dataset 类:
PyTorch模型
__init__(self)
中设置好需要的“组件"(如 conv、pooling、Linear、BatchNorm等) (3)在 forward(self, x)中用定义好的“组件”进行组装,def forward(self, x):
,x 为模型的输入nn.Sequetial
就是一个可以按照顺序封装初始化层的容器权值初始化
if isinstance(m, nn.Conv2d):
n = m.kernel_size[0] * m.kernel_size[1] * m.out_channels
m.weight.data.normal_(0, math.sqrt(2. / n))
torch
torch.sum(input, dim, out=None)
→ Tensor #与python中的sum一样print(x.sum(0))
#对一维的数求和,按列求和print(x.sum(1))
#对二维求和按行求和print(x.sum(2))
#将最小单位的数组元素相加即可new_features = super(_DenseLayer, self).forward(x)
torch.where(condition, x, y) → Tensor
对于x而言,如果其中的每个元素都满足condition,就返回x的值;如果不满足condition,就将y对应位置的元素或者y的值torch.narrow(input, dimension, start, length)
张量剪裁permute
把张量变换成不同维度,view
相当于reshape
,将元素按照行的顺序放置在新的不同大小的张量当中torch.cat(tensors, dim=0, out=None) → Tensor
将张量按照维度进行衔接torch.gesv(B, A, out=None) -> (Tensor, Tensor)
是解线性方程AX=B后得到的解torch.unsqueeze(input, dim, out=None) → Tensor
在指定位置增加一个一维的维度torch.squeeze(input, dim, out=None) → Tensor
在指定位置减去一个一维的维度,默认()就是把所有shape中为1的维度去掉detach()
就是取出一个该个tensor,并且它不会再参与梯度下降torch.nn
nn.Sequential
一个有序的容器,神经网络模块将按照在传入构造器的顺序依次被添加到计算图中执行,同时以神经网络模块为元素的有序字典也可以作为传入参数。torch.nn.functional.grid_sample(input, grid, mode='bilinear', padding_mode='zeros')
将其归一化到一个(-1,1)的二维平面上,outpuy_{x,y} 的像素值与input_{x0,y0} 的像素值一致, 矩阵索引通过grid矩阵保存。 grid_{x,y}=(x0,y0)#Sample:
import torch, torchvision
from torchsummary import summary
model = torchvision.models.vgg16()
summary(model, (3, 224, 224))
2019.7.9更新:最近接触新的论文和模型框架,梳理了一下整个学习过程:第一步是读懂论文关于整个框架的布置,知道数据是如何变化的,比如是否降分辨率,维度变化,属于分类还是拟合、聚类,找出核心和疑惑的点;第二步读懂代码在这个框架下,数据要如何预处理(需要结合特殊要求做变化),清楚核心和疑惑的点;第三步梳理整个过程,形成和完善文档;第四步改模型,调参啊敲代码之类的就是以后的事情了····
作者:pinggo
链接:https://www.pythonheidong.com/blog/article/11403/494617cefe2dccf2bea3/
来源:python黑洞网
任何形式的转载都请注明出处,如有侵权 一经发现 必将追究其法律责任
昵称:
评论内容:(最多支持255个字符)
---无人问津也好,技不如人也罢,你都要试着安静下来,去做自己该做的事,而不是让内心的烦躁、焦虑,坏掉你本来就不多的热情和定力
Copyright © 2018-2021 python黑洞网 All Rights Reserved 版权所有,并保留所有权利。 京ICP备18063182号-1
投诉与举报,广告合作请联系vgs_info@163.com或QQ3083709327
免责声明:网站文章均由用户上传,仅供读者学习交流使用,禁止用做商业用途。若文章涉及色情,反动,侵权等违法信息,请向我们举报,一经核实我们会立即删除!