![](https://img-blog.csdnimg.cn/20201014180756913.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
杭电陈冠希
这个作者很懒,什么都没留下…
展开
-
pytorch--view函数用法
在pytorch中view函数的作用为重构张量的维度,相当于numpy中resize()的功能1. torch.view(参数1,参数2…)>>> import torch>>> t1 = torch.tensor([1,2,3,4,5,6])>>> result = tt1.view(3,2)>>> resulttensor([[1, 2], [3, 4], [5, 6]])创建一个6维原创 2020-12-16 13:41:32 · 6488 阅读 · 0 评论 -
pytorch之分类问题图片数据加载
pytorch中加载数据的顺序是:①创建一个dataset对象②创建一个dataloader对象③循环dataloader对象,将data,label拿到模型中去训练1.创建datasets格式的数据这里使用的是ImageFolder,ImageFolder是一个通用的数据加载器,数据集中的数据以以下方式组织root/dog/xxx.pngroot/dog/xxy.pngroot/dog/xxz.pngroot/cat/123.pngroot/cat/nsdf3.pngroot/ca原创 2020-11-28 14:32:05 · 358 阅读 · 0 评论 -
pytorch之保存与加载模型
pytorch与保存、加载模型有关的常用函数3个:torch.save(): 保存一个序列化的对象到磁盘,使用的是Python的pickle库来实现的torch.load(): 解序列化一个pickled对象并加载到内存当中torch.nn.Module.load_state_dict(): 加载一个解序列化的state_dict对象1.state_dict在PyTorch中,一个torch.nn.Module模型的可训练参数(即权重与偏移项)保存在模型的参数(parameters,使用mod原创 2020-11-27 13:54:45 · 729 阅读 · 1 评论 -
pytorch 自动求导autograd
张量tensor张量tensor类似于numpy中的narray,是一个多维数组,两者可以相互转换 import torch import numpy as npx = np.array([1., 2., 3.],dtype=float)x = torch.from_numpy(x)注意:PyTorch的梯度只能计算浮点张量,所有从numpy转化为tensor时,需要将numpy数据格式设置为float。Autograd这个类是一个计算导数的引擎(更精确地说是雅克比向量积)。它记录原创 2020-11-29 19:34:12 · 199 阅读 · 0 评论