如何用Pytorch构建深度学习模型？

本文共计1607个文字，预计阅读时间需要7分钟。

1. 模型定义与TF相似，Pytorch通过继承父类来构建模型，同样实现两种方法。在TF中是__init__()和call()，在Pytorch中则是__init__()和forward()。功能类似，都用于初始化模型结构和定义前向传播。

1模型定义

　　和TF很像，Pytorch也通过继承父类来搭建模型，同样也是实现两个方法。在TF中是__init__()和call()，在Pytorch中则是__init__()和forward()。功能类似，都分别是初始化模型内部结构和进行推理。其它功能比如计算loss和训练函数，你也可以继承在里面，当然这是可选的。