自定义损失函数通过梯度下降在每一步进行更新
创建一个自定义层来保存可训练参数。该层不会在调用中返回输入,但是我们将使用输入来遵守如何创建层。
使用模型中的图层获取a任何输入(这与顺序模型不兼容):
a
现在,您可以尝试以某种丑陋的方式定义损失:
如果这可行,那就准备好了。
您也可以尝试使用更复杂的模型(如果您不希望a像这样在损失中使用跳过层,这可能会导致模型保存/加载问题)
在这种情况下,您将需要将其y_train作为输入而不是输出:
y_train
您的损失函数将进入一个Lambda正确接受所有参数的层:
Lambda
您的模型将输出以下损失:
您将具有虚拟损失功能:
并训练为:
你尚未登录,登录后可以
和开发者交流问题的细节
关注并接收问题和回答的更新提醒
参与内容的编辑和改进,让解决方法与时俱进
联系我
602392714
清零编程群