![](https://img-blog.csdnimg.cn/20190927151053287.png?x-oss-process=image/resize,m_fixed,h_224,w_224)
Pytorch与深度学习
文章平均质量分 59
Pytorch的学习笔记
xxxx_HeXi
这个作者很懒,什么都没留下…
展开
-
Pytorch学习笔记(五)
Pytorch学习笔记(五)一、如何使用GPU训练完整的模型测试套路结束 一、如何使用GPU训练 方法一:在网络模型,数据和损失函数处调用cuda()方法 在上一节的模型训练套路的代码中直接进行修改 #准备数据集 train_data = torchvision.datasets.CIFAR10(root="./dataset", train=True, transform=torchvision.transforms.ToTensor(), download=True) test_data = torch原创 2021-11-09 21:35:41 · 740 阅读 · 0 评论 -
Pytorch学习笔记(四)
Pytorch学习笔记(四)一、损失函数与反向传播1、L1Loss2、MSELoss3、CrossEntropyLoss4、将loss应用到神经网络中5、反向传播优化器二、网络模型1、现有网络模型的使用和修改2、网络模型的保存与读取完整的模型训练套路 一、损失函数与反向传播 1、L1Loss 结果为输出与标签之间的差的和平均值或是和 inputs = torch.tensor([1,2,3], dtype = torch.float32) targets = torch.tensor([1,2,5],dty原创 2021-11-09 20:15:50 · 599 阅读 · 0 评论 -
Pytorch学习笔记(三)
Pytorch学习笔记(三)一、torch.nn1、Module2、Conv2d3、Pooling layers——MaxPool2d4、非线性激活函数——Relu5、线性层和其它层6、搭建神经网络以及Sequential层 一、torch.nn 这里不讲解关于卷积神经网络的内容了,对这方面不了解的朋友可以自己搜索学习一下啦! 1、Module 使用一个例子来说明如何使用Module class Test(nn.Module): #创建的类需要继承Module类 def __init__(self):原创 2021-11-09 09:56:07 · 538 阅读 · 0 评论 -
Pytorch学习笔记(二)
Pytorch学习笔记(二)Torchvision数据集torchvision.datasetsDataLoader结束 Torchvision数据集 数据集可以在Pytorch官网上下载 torchvision.datasets 参数: root(数据集存放在什么位置) train(数据集是训练集还是测试集,输入为bool类型) transform(对数据集当中的所有数据进行什么样的变换 ) target_transform download(是否在网上下载数据集,输入为bool类型,如果数据集下载慢可以原创 2021-11-08 15:29:12 · 85 阅读 · 0 评论 -
Pytorch学习笔记(一)
Pytorch学习笔记(一)Pytorch加载数据DatasetDataloaderTensorBoard的使用SummaryWriter类的使用add_scalaradd_imageTransforms的使用ToTensor()Normalize()Resize()Compost()RandomCrop()总结 Pytorch加载数据 Dataset 作用:提供一种方式获取数据和label并编号,可以根据编号提取相应的数据,同时获取数据对应的相应的label class MyData(Dataset)原创 2021-11-06 22:52:07 · 1649 阅读 · 0 评论