Skip to content

Commit 21993ee

Browse files
committed
keeping empty tensor for fgradInput in ConvMM
nilling the fgradInput in ConvolutionMM causes failure in post-training
1 parent cc09260 commit 21993ee

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

train-face-detector/train.lua

+1-1
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@ function nilling(module)
3131
if module.finput then module.finput = torch.Tensor() end
3232
module.gradWeight = nil
3333
module.output = torch.Tensor()
34-
module.fgradInput = nil
34+
if module.fgradInput then module.fgradInput = torch.Tensor() end
3535
module.gradInput = nil
3636
end
3737

0 commit comments

Comments
 (0)