![](https://img-blog.csdnimg.cn/20201014180756916.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
sjtu_leexx
这个作者很懒,什么都没留下…
展开
-
pytorch建立自己的数据集(以mnist为例)
本文将原始的numpy array数据在pytorch下封装为Dataset类的数据集,为后续深度网络训练提供数据。#加载并保存图像信息首先导入需要的库,定义各种路径。import osimport matplotlibfrom keras.datasets import mnistimport numpy as npfrom torch.utils.data.dataset imp...原创 2018-10-12 19:26:01 · 16143 阅读 · 3 评论 -
用pytorch搭建AlexNet(微调预训练模型及手动搭建)
本文介绍了如何在pytorch下搭建AlexNet,使用了两种方法,一种是直接加载预训练模型,并根据自己的需要微调(将最后一层全连接层输出由1000改为10),另一种是手动搭建。构建模型类的时候需要继承自torch.nn.Module类,要自己重写__\_\___init__\_\___方法和正向传递时的forward方法,这里我自己的理解是,搭建网络写在__\_\___init__\_\___...原创 2018-10-14 15:58:42 · 16434 阅读 · 5 评论 -
pytorch搭建网络测试时报错Expected object of type torch.DoubleTensor but found type torch.FloatTensor...
用pytorch搭建网络测试时,代码报错如下:Expected object of type torch.DoubleTensor but found type torch.FloatTensor for argument #2 ‘weight’搭建的网络为AlexNet,测试代码如下x = np.random.rand(1,3,224,224)# x=x.astype(np.flo...原创 2018-10-14 16:03:59 · 5402 阅读 · 1 评论 -
pytorch下搭建网络训练并保存模型
最近在学习pytorch,使用mnist数据集,搭建AlexNet训练并保存模型,将代码做一记录。建立数据集的方法见pytorch建立自己的数据集(以mnist为例)搭建网络的方法见用pytorch搭建AlexNet(微调预训练模型及手动搭建)训练代码如下:import torchimport osfrom torchvision import transformsimport to...原创 2018-10-18 13:56:12 · 4209 阅读 · 0 评论 -
pytorch criterion踩坑小结
1. 数据类型不匹配:报错:Expected object of type torch.LongTensor but found type torch.FloatTensor for argument #2 ‘target’criterion = nn.CrossEntropyLoss()loss = criterion(y_pre, y_train)这里的y_train类型一定要是Lo...原创 2018-10-17 16:17:59 · 31299 阅读 · 7 评论 -
pytorch下只打印tensor的数值不打印出device等信息的方法
torch.Tensor类型的数据loss和acc打印时 ,如果写成以下写法print('batch_loss: '+str(loss.data)+'batch acc: '+str(acc.data))则不仅会打印出loss和acc的值,还会打印出device信息和 tensor字样,如下:如果仅想打印出数值,使得打印出的信息更加简洁,则要用以下写法print('batch_loss...原创 2018-10-17 19:32:49 · 22890 阅读 · 4 评论