【深度学习】Pytorch四步阅读法

Pytorch框架快速入门

前言

找数据定义、找model定义、(找损失函数、优化器定义),主循环代码逻辑

Pytorch框架编写深度学习模型,无非就是按照四步,从这四步出发去剖析和书写代码,可以比较好的提高我们的效率。

基本思想和步骤

img

  • 四大步骤

1、输入处理模块 (X 输入数据,变成网络能够处理的Tensor类型)

2、模型构建模块 (主要负责从输入的数据,得到预测的y^, 这就是我们经常说的前向过程)

3、定义代价函数和优化器模块 (注意,前向过程只会得到模型预测的结果,并不会自动求导和更新,是由这个模块进行处理)

4、构建训练过程 (迭代训练过程,就是上图表情包的训练迭代过程

实例讲解

image-20200625181622497

猜你喜欢

转载自www.cnblogs.com/YiXinLiu617/p/13198288.html