我想使用本教程nsl.keras.GraphRegularization
中描述的方法来训练二元目标深度神经网络模型。我的模型在不应该“图形正则化”的中间密集层中有一个三元组半硬损失。
根据Github 上的nsl.keras.GraphRegularization
定义:
将图正则化合并到 的损失中
base_model
。图正则化在 logits 层上完成,并且仅在训练期间完成。
是说中间三元组半硬损失不会受到这个正则化的影响吗?
是的,这是正确的。图正则化将仅应用于 的输出base_model
。如果您base_model
在另一层中使用三元组半硬损失,则该损失应保持不受影响并保留。如果不是这样,请在https://github.com/tensorflow/neural-structured-learning/issues提交错误。