lua - 在 Torch 中添加自定义损失函数

Question

在 Torch 中实现自定义损失函数需要哪些必要步骤？

看来您必须为 updateOutput 和 updateGradInput 编写一个实现。

就这些了吗？那么你基本上创建了一个新类:

local CustomCriterion, parent =   torch.class('CustomCriterion','nn.Criterion')

并实现以下两个功能:

function CustomCriterion:updateOutput(input, target)
function CustomCriterion:updateGradInput(input, target)

这是正确的吗？还是还有更多工作要做？

此外，对于所提供的标准，这些函数是用 C 实现的，但我认为 Lua 实现也可以工作，尽管可能会慢一点？

Answer 1

我已经实现了以下形式的函数(以伪代码形式)

--assuming input is partitioned in input_a,input_b
--         target is accordingly partitionend in target_a, target_b  
f(input)=MSE(input_a,target_a)+ custom_sutff(input_b,target_b)

就像你描述的那样，有好几次了。所以，据我所知，我认为你的两个问题的答案都是肯定的。

基本上nn/MSECriterion.lua和 this似乎支持这一点。