From 8065ec9bb2d764bc3a26f751623049ec813d869e Mon Sep 17 00:00:00 2001 From: nagadomi Date: Sat, 13 Oct 2018 20:06:12 +0000 Subject: [PATCH] fix --- lib/AuxiliaryLossCriterion.lua | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lib/AuxiliaryLossCriterion.lua b/lib/AuxiliaryLossCriterion.lua index 63170fd..81c9682 100644 --- a/lib/AuxiliaryLossCriterion.lua +++ b/lib/AuxiliaryLossCriterion.lua @@ -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