diff options
Diffstat (limited to 'torch/legacy/nn/ParallelCriterion.py')
-rw-r--r-- | torch/legacy/nn/ParallelCriterion.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/torch/legacy/nn/ParallelCriterion.py b/torch/legacy/nn/ParallelCriterion.py index a44541caab..7ecfd95c6b 100644 --- a/torch/legacy/nn/ParallelCriterion.py +++ b/torch/legacy/nn/ParallelCriterion.py @@ -2,6 +2,7 @@ import torch from .Criterion import Criterion from .utils import recursiveResizeAs, recursiveFill, recursiveAdd + class ParallelCriterion(Criterion): def __init__(self, repeatTarget=False): @@ -36,4 +37,3 @@ class ParallelCriterion(Criterion): def type(self, type=None, tensorCache=None): self.gradInput = [] return super(ParallelCriterion, self).type(type, tensorCache) - |