栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

在pytorch深度学习中,损失函数和优化器的区别,一看就懂系列

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

在pytorch深度学习中,损失函数和优化器的区别,一看就懂系列

A,首先看代码

def loss_fn(y,y_predict):
    loss = (y_predict-y).pow(2).mean()
    for i in (w, b):
        if i.grad is not None:
            i.grad.data.zero_()  #or   optimizer.zero_grad()
    loss.backward()
    return loss.data


def optimize(learning_rate):
    w.data -= learning_rate*w.grad.data
    b.data -= learning_rate*b.grad.data

B,看到代码,我们更加清楚了,这个是是手写的优化器就是每次调整参数的方法,损失函数就是如何计算输入和输出之间的损失差值
C,在正常开发的过程中,我们都是用模型直接调用,很少手工写这些,例如SGD作为随即梯度下降的优化器;MESSLoss均方误差计算损失函数 的模块,直接调用就可以了。

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/700968.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号