https://github.com/fqnchina/CEILNet
Raw File
Tip revision: 80e46959e14f168aa4bc0f4faafdfb5ebfee3821 authored by Qingnan Fan on 11 September 2018, 09:23:36 UTC
Update README.md
Tip revision: 80e4695
L1Criterion.lua
local L1Criterion, parent = torch.class('nn.L1Criterion', 'nn.Criterion')

function L1Criterion:__init()
   parent.__init(self)
end

function L1Criterion:updateOutput(input, target)
   self.output = torch.mean(torch.abs(torch.csub(input,target)))
   return self.output
end

function L1Criterion:updateGradInput(input, target)
   local num = input:size(1)*input:size(2)*input:size(3)*input:size(4)
   self.gradInput = torch.CudaTensor():resizeAs(input):fill(-1)
   self.gradInput[torch.ge(torch.csub(input,target),0)] = 1
   self.gradInput = self.gradInput/num
   return self.gradInput
end
back to top