Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

SpatialSubtractiveNormalization:clearState() missing several cleanups #1303

Open
wants to merge 11 commits into
base: master
Choose a base branch
from
Open

SpatialSubtractiveNormalization:clearState() missing several cleanups #1303

Changes from all commits
Commits
Show all changes
11 commits
Select commit Hold shift + click to select a range
cd4ba2c
Merge pull request #1 from torch/master
elikosan Dec 20, 2016
e52775a
Merge remote-tracking branch 'refs/remotes/torch/master'
elikosan Mar 2, 2017
55e1eee
Merge pull request #2 from torch/master
elikosan May 18, 2017
50c18ef
Merge pull request #3 from torch/master
elikosan Jun 1, 2017
d09f71a
replace lon glong types with size_t
elikosan Jul 25, 2017
b459f84
Merge pull request #4 from elikosan/2017-06-01
elikosan Jul 27, 2017
27ab9ea
Merge pull request #5 from torch/master
elikosan Oct 2, 2017
93b646c
fix condition logic for coef recalculation in SpatialSubtractiveNorm…
elikosan Oct 2, 2017
7be70a6
SpatialSubtractiveNormalization:clearState: clear cached size to make…
elikosan Oct 2, 2017
09b536a
Merge pull request #6 from torch/master
elikosan Oct 2, 2017
d095326
SpatialSubtractiveNormalization:clearState missing several cleanups
elikosan Jan 2, 2018
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
10 changes: 6 additions & 4 deletions SpatialSubtractiveNormalization.lua
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -109,8 +109,10 @@ function SpatialSubtractiveNormalization:updateGradInput(input, gradOutput)
end

function SpatialSubtractiveNormalization:clearState()
if self.ones then self.ones:set() end
if self._coef then self._coef:set() end
self.meanestimator:clearState()
return parent.clearState(self)
nn.utils.clear(self, '_inpsz', 'ones', '_coef')
self.coef = torch.Tensor(1,1,1)
self.subtractor:clearState()
self.divider:clearState()
self.meanestimator:clearState()
return parent.clearState(self)
end