lua - 重写 torch 中的 updateGradInput 方法不适用于我的自定义模块

Question

我实现了一个名为sharedModule 的模块供我自己使用。我已经重写了下面的 updateGradInput 方法:

    function SharedModule:updateGradInput(input, gradOutput)
    test_grad = {}

    print("call updateGradInput")
    test_input = input
    test_gradOutput = gradOutput
    assert(type(gradOutput) == 'table' and #input == #gradOutput)

    local T = #input
    for t = 1, T do
            self.gradInput[t] = self.clones[t]: updateGradInput(input[t], gradOutput[t])
            test_grad[t] = self.gradInput[t]
    end
    print(#self.gradInput)  -- print normal value 
    --self.gradInput = test_grad  -- 
    return self.gradInput  -- empty, ???
    end

但是，当我在模块上调用向后方法时，self.gradInput 字段没有更新，这是什么问题，任何人都可以帮助我吗？

Answer 1

backward方法会调用两个名为updateGradInput和accGradParams的方法，结果发现accGradParams由于拼写错误而出现错误