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

咱们关心网络结构和数据,定义损失函数,定义优化函数等。算法 具体步骤以下:网络 第一步:图像封装为向量后,将输入input向前传播,进行运算后获得输出outputdom 第二步:将output再输入loss函数,计算loss值(是个标量)-损失函数用来获得新权重函数 第三步:将梯度反向传播到每一个参数(优化函数):主要指标是 学习速率𝜂  x 梯度向量g学习 第四步:利用下面公式进行权重更新优化
相关文章
相关标签/搜索