0

我想使用本教程nsl.keras.GraphRegularization中描述的方法来训练二元目标深度神经网络模型。我的模型在不应该“图形正则化”的中间密集层中有一个三元组半硬损失。

根据Github 上的nsl.keras.GraphRegularization 定义:

将图正则化合并到 的损失中base_model

图正则化在 logits 层上完成,并且仅在训练期间完成。

是说中间三元组半硬损失不会受到这个正则化的影响吗?

4

1 回答 1

0

是的,这是正确的。图正则化将仅应用于 的输出base_model。如果您base_model在另一层中使用三元组半硬损失,则该损失应保持不受影响并保留。如果不是这样,请在https://github.com/tensorflow/neural-structured-learning/issues提交错误。

于 2019-12-04T22:23:05.777 回答