您好, 欢迎来到 !    登录 | 注册 | | 设为首页 | 收藏本站

自定义损失函数通过梯度下降在每一步进行更新

自定义损失函数通过梯度下降在每一步进行更新

创建一个自定义层来保存可训练参数。该层不会在调用中返回输入,但是我们将使用输入来遵守如何创建层。

使用模型中的图层获取a任何输入(这与顺序模型不兼容):

现在,您可以尝试以某种丑陋的方式定义损失:

如果这可行,那就准备好了。

您也可以尝试使用更复杂的模型(如果您不希望a像这样在损失中使用跳过层,这可能会导致模型保存/加载问题)

在这种情况下,您将需要将其y_train作为输入而不是输出

您的损失函数将进入一个Lambda正确接受所有参数的层:

您的模型将输出以下损失:

您将具有虚拟损失功能

并训练为:

其他 2022/1/1 18:30:09 有506人围观

撰写回答


你尚未登录,登录后可以

和开发者交流问题的细节

关注并接收问题和回答的更新提醒

参与内容的编辑和改进,让解决方法与时俱进

请先登录

推荐问题


联系我
置顶