小玺大大

上一节，我们给出了模型加载和保存的简要示例，但是，我们有时候会用别人的参数，他们的层参数名和我们的名称很容易不同，因此这里将会对源码进入深入剖析，分析参数提取和保存是如何实现的。1 模型文件分析我们使用pytorch的VGG16预训练模型，加载，返回其类型。可以发现，是OrderedDict类型，也就是字典类型，既然是字典，每个层的参数就是用了一个键值对保存起来了。model = torch...

前4个博客，已经介绍了什么是Tensor，构建模型所需的基本类有哪些，这些类有哪些工作。下面，结合这些内容，通过简单的例子进行使用说明。文章目录1 构造模块化的神经网络2 损失函数3 训练方法3.1 SGD方法3.2 Adam方法3.3 简单用法4 网络模型库 torchvision.models5 模型加载与保存5.1 模型加载5.1.1 加载自带的预训练模型5.1.2 加载本地的训练模型5...

下面给出模型基本类相关的源码分析。本节大部分函数重载用法与上节内容相似，在本节一样的内容将不会进行详细描述文章目录1 nn.Sequential2 nn.ModuleList3 nn.ModuleDict总结1 nn.Sequential这个类可以快速的构建一个模型，下面是官方给的一个类。model = nn.Sequential( nn.Conv2d(1,20,5)...

为了更好理解Pytorch基本类的实现方法，我这里给出了关于参数方面的3个类的源码详解。此部分可以更好的了解实现逻辑结构，有助于后续代码理解，学pytorch的话这个不是必须掌握的，看不懂也没关系。文章目录1 Parameter 参数类源码2 ParameterList 参数列表类源码3 ParameterDict 参数字典类源码总结1 Parameter 参数类源码此部分参考《pyto...

结合博客《Pytorch学习(1) —— Tensor基础》我们已经了解了Tensor的一些基础知识，下面开始结合这些介绍TORCH.NN里面的一些内容，这些内容包含了构建网络的一些基本类(Module, Prameters等)操作，并对一些关键网络函数(卷积，池化等)进行整理，作为字典方便后续查询。正常来说，Tensor说完应该介绍torch.autograd自动求导问题，但是在具体应用中，似...

学习时候作为Pytorch的学习笔记，对一些关键函数进行整理，方便后续学习时候查询用。许多教程基于老版本Pytorch，本博客基于1.*系列的版本本文章还在更新中，目前仅更新常用的一些方法文章目录1 Tensor的创建1.1 使用torch.tensor创建1.2 其他创建操作（比如全0，全1，随机等等）Tensor 维度，访问1 Tensor的创建注意区分: torch.Tensor是...