diff options
author | Arjun Jain <arjunjain@gmail.com> | 2014-12-18 02:29:27 +0300 |
---|---|---|
committer | Arjun Jain <arjunjain@gmail.com> | 2014-12-18 02:29:27 +0300 |
commit | 2c6d601f3a76931bfb950263c8cb4d9689a5fe00 (patch) | |
tree | f2810fa0f6619e29a20cd1f1c713fcce7f31330e | |
parent | e71e246e172b757f81b5da4a58871104bc81d3d9 (diff) |
Temporary buffers for unfolding (for CUDA version)
-rw-r--r-- | VolumetricConvolution.lua | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/VolumetricConvolution.lua b/VolumetricConvolution.lua index 4dec9e3..aea94a5 100644 --- a/VolumetricConvolution.lua +++ b/VolumetricConvolution.lua @@ -20,7 +20,9 @@ function VolumetricConvolution:__init(nInputPlane, nOutputPlane, kT, kW, kH, dT, self.bias = torch.Tensor(nOutputPlane) self.gradWeight = torch.Tensor(nOutputPlane, nInputPlane, kT, kH, kW) self.gradBias = torch.Tensor(nOutputPlane) - + -- temporary buffers for unfolding (CUDA) + self.finput = torch.Tensor() + self.fgradInput = torch.Tensor() self:reset() end |