diff options
Diffstat (limited to 'torch/legacy/nn/Identity.py')
| -rw-r--r-- | torch/legacy/nn/Identity.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/torch/legacy/nn/Identity.py b/torch/legacy/nn/Identity.py index ed9f302f28..09c7e1b8e8 100644 --- a/torch/legacy/nn/Identity.py +++ b/torch/legacy/nn/Identity.py @@ -2,6 +2,7 @@ import torch from .Module import Module from .utils import clear + class Identity(Module): def updateOutput(self, input): @@ -14,8 +15,7 @@ class Identity(Module): def clearState(self): clear(self, [ - 'output', - 'gradInput', + 'output', + 'gradInput', ]) return super(Identity, self).clearState() - |