
源码解析
Godswisdom
这个作者很懒,什么都没留下…
展开
-
Pytorch模型训练(4) - Loss Function
本文总结Pytorch中的Loss Function原创 2019-01-24 16:39:04 · 8836 阅读 · 0 评论 -
Pytorch模型训练(5) - Optimizer
本文总结Pytorch中的Optimizer Optimizer是深度学习模型训练中非常重要的一个模块,它决定参数参数更新的方向,快慢和大小,好的Optimizer算法和合适的参数使得模型收敛又快又准 但本文不会讨论什么任务用什么Optimizer,及其参数设置,只是总结下Pytorch中的Optimizer...原创 2019-01-25 17:05:29 · 11805 阅读 · 0 评论 -
Pytorch模型训练(0) - CPN源码解析
本人在博客2D多人姿态估计–《Cascaded Pyramid Network for Multi-Person Pose Estimation》中简述来CPN的原理,这里则来解析下其源码;论文作者提供的源码是tensorflow版本的,之前有人分析过了;正好还有个pytorch版本的CPN 本人想通过这份源码的学习来总结学习Pytorch模型训练的一些相关知识;pytorch-cpn这份源码...原创 2019-01-25 17:13:33 · 4732 阅读 · 28 评论 -
Pytorch模型训练(1) - 模型定义
本文来总结Pytorch训练中的模型结构一些内容,包括模型的定义,模型参数化初始化方法,模型的保存与加载等1 模型定义–三要素 1)Pytorch模型的定义都需要继承nn.module这个类 读者可以在Pytorch源码下/torch/nn/modules/module.py中找到这个类的实现:class Module(object): r""&原创 2019-01-21 17:24:01 · 14890 阅读 · 7 评论 -
Pytorch模型训练(6) - 数据加载
前面几篇博客分析Pytorch模型训练的模型定义,损失函数及优化器,本文则来看看在模型训练中,又一非常重要的模块—数据加载 在深度学习模型训练中,我们面对的训练任务是多种多样的,不同任务面对的数据格式也是不同的,甚至相同任务,也会面对不同格式的数据集;所以不存在所谓的通用数据脚本,只能是具体任务,具体数据集格式,单独对待。 但是呢!!!深度学习框架一般都会为数据加载提供同一的接口,我们...原创 2019-01-28 16:13:12 · 3770 阅读 · 0 评论 -
Pytorch模型训练(2) - 模型初始化
本系列来总结Pytorch训练中的模型结构一些内容,包括模型的定义,模型参数化初始化方法,模型的保存与加载等原创 2019-01-23 11:38:28 · 6906 阅读 · 1 评论 -
Pytorch模型训练(3) - 模型保存与加载
本系列来总结Pytorch训练中的模型结构一些内容,包括模型的定义,模型参数化初始化方法,模型的保存与加载等1 保存和加载 1)首先看Save源码 save使用pickle工具将模型对象序列化为pickle文件到diskdef save(obj, f, pickle_module=pickle, pickle_protocol=DEFAULT_PROTOCOL): ""&am原创 2019-01-23 11:39:25 · 18857 阅读 · 0 评论