diff options
| author | orian <pawelszczur@gmail.com> | 2015-05-27 02:59:28 +0300 |
|---|---|---|
| committer | orian <pawelszczur@gmail.com> | 2015-05-27 02:59:28 +0300 |
| commit | 4bc5508408b12bf5cc82e743309f60ffdac12073 (patch) | |
| tree | b7f915d3ef4b541036cae39226867482e6775d5d /SpatialConvolution.lua | |
| parent | 1cca535acfe22fb68dd060a2544a1abc44fdbab3 (diff) | |
Add more useful string description of layers.
Adds:
- pbb in Dropout and SpatialDropout
- non default params in SpatialConvolution, SpatialConvonlutionMM, SpatialMaxPooling
Diffstat (limited to 'SpatialConvolution.lua')
| -rw-r--r-- | SpatialConvolution.lua | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/SpatialConvolution.lua b/SpatialConvolution.lua index 4a42b63..c5fce40 100644 --- a/SpatialConvolution.lua +++ b/SpatialConvolution.lua @@ -119,3 +119,15 @@ function SpatialConvolution:type(type) self.fgradInput = torch.Tensor() return parent.type(self,type) end + +function SpatialConvolution:__tostring__() + local s = string.format('%s(in: %d, out: %d, kW: %d, kH: %d', torch.type(self), + self.nInputPlane, self.nOutputPlane, self.kW, self.kH) + if self.dW ~= 1 or self.dH ~= 1 then + s = s .. string.format(', dW: %d, dH: %d', self.dW, self.dH) + end + if self.padding ~= 0 then + s = s .. ', padding: ' .. self.padding + end + return s .. ')' +end |