实战人工智能框架PyTorch的模型训练

我们关心网络结构和数据，定义损失函数，定义优化函数等。 具体步骤如下： 第一步：图像封装为向量后，将输入input向前传播，进行运算后得到输出output 第二步：将output再输入loss函数，计算loss值（是个标量）-损失函数用来得到新权重 第三步：将梯度反向传播到每个参数（优化函数）：主要指标是 学习速率𝜂  x 梯度向量g 第四步：利用下面公式进行权重更新 新权重w =  旧权重w 

>>阅读原文<<