CodeMirror/waifu2x
1
0
Fork 0
mirror of synced 2024-05-17 03:12:18 +12:00
Code Issues Releases Wiki Activity

fix

Browse source
This commit is contained in:
nagadomi 2018-10-13 20:06:12 +00:00
parent e6da46d08f
commit 8065ec9bb2
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
lib/AuxiliaryLossCriterion.lua
View file

@ -37,7 +37,7 @@ function AuxiliaryLossCriterion:updateOutput(input, target)
else
self.criterions[1] = self.base_criterion()
end
self.criterions[1].sizeAverage = self.sizeAverage()
self.criterions[1].sizeAverage = self.sizeAverage
if input:type() == "torch.CudaTensor" then
self.criterions[1]:cuda()
end