CodeMirror/waifu2x
1
0
Fork 0
mirror of synced 2024-06-13 16:34:32 +12:00
Code Issues Releases Wiki Activity

Fix gradient scale in WeightedMSECriterion

Browse source
This commit is contained in:
nagadomi 2015-10-31 04:35:33 +09:00
parent 4c691b4640
commit 9d63ed2947
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
lib/WeightedMSECriterion.lua
View file

@ -19,6 +19,7 @@ function WeightedMSECriterion:updateOutput(input, target)
end
function WeightedMSECriterion:updateGradInput(input, target)
self.gradInput:resizeAs(input):copy(self.diff)
local norm = 2.0 / input:nElement()
self.gradInput:resizeAs(input):copy(self.diff):mul(norm)
return self.gradInput
end