pytorch深度学习基本流程概要

最新推荐文章于 2024-07-31 00:21:09 发布

傅里叶算计我

最新推荐文章于 2024-07-31 00:21:09 发布

阅读量311

点赞数 1

文章标签：深度学习人工智能 python 神经网络

本文链接：https://blog.youkuaiyun.com/qq_40264206/article/details/109096071

版权

在pytorch中，进行深度学习时，有四大模块：
Data（数据）、 Module（模型）、Loss function（损失函数）、Optimizer（优化器）
Data分为inputs和targets

基本流程：
(1)inputs输入到Module中，会得到一个预测值(outputs)
(2)outputs、targets放置到Loss function中会得到Loss
(3)使用loss.backward后，会得到参数的梯度值(grad)
(4)optimizer拿到grad后，会更新函数的参数