diff options
| author | Tatsunosuke Shimada <adam.is.t521@gmaill.com> | 2015-07-15 03:53:39 +0300 |
|---|---|---|
| committer | Tatsunosuke Shimada <adam.is.t521@gmaill.com> | 2015-07-15 03:53:39 +0300 |
| commit | 88abb3dd57b484406bc1eabef8910b90501ae461 (patch) | |
| tree | 1f60ccfaea8adc72f40bf6abb8dd4f2de34db3cb /SpatialDropout.lua | |
| parent | 9a4155c0a966464a8b66daf66266ba9f47f5e67a (diff) | |
small fix on error message
Diffstat (limited to 'SpatialDropout.lua')
| -rw-r--r-- | SpatialDropout.lua | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/SpatialDropout.lua b/SpatialDropout.lua index 094e5bd..37680d4 100644 --- a/SpatialDropout.lua +++ b/SpatialDropout.lua @@ -15,7 +15,7 @@ function SpatialDropout:updateOutput(input) elseif input:dim() == 3 then self.noise:resize(input:size(1), 1, 1) else - error('Input must be 4D (nbatch, nfeat, h, w) or 2D (nfeat, h, w)') + error('Input must be 4D (nbatch, nfeat, h, w) or 3D (nfeat, h, w)') end self.noise:bernoulli(1-self.p) -- We expand the random dropouts to the entire feature map because the |