diff options
| author | Tongzhou Wang <SsnL@users.noreply.github.com> | 2018-02-02 12:39:33 -0500 |
|---|---|---|
| committer | Soumith Chintala <soumith@gmail.com> | 2018-02-02 12:39:33 -0500 |
| commit | f23feca681c5066c70f0fe1516fc2e269d615e93 (patch) | |
| tree | 6af46262286c9db155b302df9f14ede4f06b7ae2 /torch/legacy | |
| parent | e22095b09d19f2b9aa964ea10139eb11340624fd (diff) | |
| download | pytorch-f23feca681c5066c70f0fe1516fc2e269d615e93.tar.gz pytorch-f23feca681c5066c70f0fe1516fc2e269d615e93.tar.bz2 pytorch-f23feca681c5066c70f0fe1516fc2e269d615e93.zip | |
Fix output_nr not incremented correctly (#4812)
* fix output_nr not incremented correctly
* update test_conv_double_backward to cover this case; call accGradParameters if any param (not just weight) requires grad in parse_nn.py
* update Spatial/VolumetricFull(Dilated)Convolution to support accGradParameters with only bias requiring grad
* Spatial/VolumetricConvolutionMM
* Spatial/VolumetricDilatedConvolution
* address @fmassa 's comments
Diffstat (limited to 'torch/legacy')
| -rw-r--r-- | torch/legacy/nn/VolumetricFullConvolution.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/torch/legacy/nn/VolumetricFullConvolution.py b/torch/legacy/nn/VolumetricFullConvolution.py index 18b7172d13..e0eb3faed2 100644 --- a/torch/legacy/nn/VolumetricFullConvolution.py +++ b/torch/legacy/nn/VolumetricFullConvolution.py @@ -102,6 +102,7 @@ class VolumetricFullConvolution(Module): self.bias, self.finput, self.fgradInput, + self.kT, self.kW, self.kH, self.dT, self.dW, self.dH, self.padT, self.padW, self.padH, adjT, adjW, adjH @@ -138,6 +139,7 @@ class VolumetricFullConvolution(Module): self.weight, self.finput, self.fgradInput, + self.kT, self.kW, self.kH, self.dT, self.dW, self.dH, self.padT, self.padW, self.padH, adjT, adjW, adjH @@ -179,6 +181,7 @@ class VolumetricFullConvolution(Module): self.gradBias, self.finput, self.fgradInput, + self.kT, self.kW, self.kH, self.dT, self.dW, self.dH, self.padT, self.padW, self.padH, adjT, adjW, adjH, |