![](https://img-blog.csdnimg.cn/20201014180756922.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch入门
黎明静悄悄啊
这个作者很懒,什么都没留下…
展开
-
基于pytorch下一个简单的模型运用过程
将训练好的模型进行应用,值得注意的是需要调用odel.eval()接口,取消掉网路的dropout层。并且,该过并不需要逆向传播。model=NeuralNetwork()model.load_state_dict(torch.load("model.path"))class=["T-shirt/top", "Trouser", "Pullover", "Dress", "Coat", "Sandal", "Shirt", "Sneaker",原创 2021-08-18 10:10:56 · 112 阅读 · 0 评论 -
基于pytorch框架下的一个简单的train与test代码
个人认为最大的差异有两点:1,train需要使用到batch概念,test并不需要使用到batch概念。2,train需要使用逆向传播,在逆向传播中更改W权重数值,test并不需要逆向传播。def train(dataloader,model,loss_fn,optimizer): size=len(dataloader.dataset) for batch,(X,y) in enumerate(dataloader): X,y=X.to(device)原创 2021-08-18 10:01:09 · 1151 阅读 · 0 评论 -
索引与切片
indexingselcet by steping具体的索引采样select by mask使用的并不多原创 2020-08-30 14:57:44 · 132 阅读 · 0 评论 -
创建tensor
从numpy中引入:从list中导入:tensor小写是接收数据,大写是接受维度。申请一片空间不需要初始化:数据的初始化:随机初始化:原创 2020-08-29 10:41:25 · 77 阅读 · 0 评论 -
pytorch张量的数据类型
pytorch的数据类型:类型的检验:生成维度为0的tensor:维度为1的 平常我们称为向量 但是这里我们成为张量。,tensor()直接给出数据,.Floattensor()给出数据的维度。原创 2020-08-27 13:24:46 · 139 阅读 · 0 评论 -
Lenet网络实现
import torchfrom torch import nnfrom torch.nn import functional as Fclass Lenet5(nn.Module): #定义Lenet网络结构类 """ for cifar10 dataset. """ def __init__(self): super(Lenet5, self).__init__() #调用类的方法初始化父类 self.conv_u原创 2020-08-21 13:15:04 · 241 阅读 · 0 评论 -
pytorch入门2 常用函数
import numpy as npimport torch#获取张量数据类型a = torch.randn(2, 3) #返回-1到1之间的随机数2行3列print(a)b = a.shape #返回a形状 touch.Size([2,3])print(b)c = a.size() #返回的值和shape是一样的d = a.size(1) #放回size(【2,3】)中第二个元素 即3,a.shape(1)返回的值一样a = torch.rand(2, 2, 3) #创建一个3维原创 2020-08-21 09:10:10 · 143 阅读 · 0 评论 -
pytorch入门1
我们先来对比一下pytorch之间使用GPU与CPU之间的差距import torchimport timeprint(torch.__version__) #查看版本print(torch.cuda.is_available()) #能否运行gpu版本cuda# print('hello, world.')a = torch.randn(10000, 1000)b = torch.randn(1000, 2000)t0 = time.time()c =原创 2020-08-21 08:25:00 · 116 阅读 · 0 评论