CodeMirror/waifu2x
1
0
Fork 0
mirror of synced 2024-05-16 19:02:21 +12:00
Code Issues Releases Wiki Activity

Fix a bug in ClippedMSECriterion:backward

Browse source
This commit is contained in:
nagadomi 2017-01-09 12:55:50 +09:00
parent 742c9fdd95
commit 6051dad793
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
lib/ClippedMSECriterion.lua
View file

@ -5,12 +5,14 @@ function ClippedMSECriterion:__init(min, max)
self.min = min
self.max = max
self.diff = torch.Tensor()
self.diff_pow2 = torch.Tensor()
end
function ClippedMSECriterion:updateOutput(input, target)
self.diff:resizeAs(input):copy(input)
self.diff:clamp(self.min, self.max)
self.diff:add(-1, target)
self.output = self.diff:pow(2):sum() / input:nElement()
self.diff_pow2:resizeAs(self.diff):copy(self.diff):pow(2)
self.output = self.diff_pow2:sum() / input:nElement()
return self.output
end
function ClippedMSECriterion:updateGradInput(input, target)