我想在 MatConvNet 中实现一个自定义损失函数(来自本文:https : //arxiv.org/abs/1706.00909)。
我的代码使用 DagNN 包装器。我想修改类SegmentationLoss()
以使用自定义损失函数,custom_loss()
而不是vl_nnloss()
. 对于forward()
通过,custom_loss()
返回计算的损失值。
我不明白的是在传递过程中custom_loss()
应该做什么。什么是额外的输入,它来自哪里,返回值应该是什么?backward()
SegmentationLoss()
derOutputs
custom_loss()
谢谢!