Skip to content
Browse files

Better fovea print

  • Loading branch information...
1 parent 49c569b commit 6402f6b13f5c8a4c3adcbbe64889531666af4fde @clementfarabet committed Jan 26, 2012
Showing with 5 additions and 5 deletions.
  1. +5 −5 SpatialFovea.lua
View
10 SpatialFovea.lua
@@ -373,16 +373,16 @@ function SpatialFovea:__tostring__()
local line = '\n'
local next = ' |`-> '
local ext = ' | '
- local extlast = ' '
local last = ' ... -> '
local str = 'nn.SpatialFovea'
str = str .. ' {' .. line .. tab .. 'input'
for i=1,#self.processors do
- if i == self.processors then
- str = str .. line .. tab .. next .. '(' .. i .. '): ' .. tostring(self.processors[i]):gsub(line, line .. tab .. extlast)
- else
- str = str .. line .. tab .. next .. '(' .. i .. '): ' .. tostring(self.processors[i]):gsub(line, line .. tab .. ext)
+ local pipeline = nn.Sequential()
+ if self.preProcessors[i] then
+ pipeline:add(self.preProcessors[i])
end
+ pipeline:add(self.processors[i])
+ str = str .. line .. tab .. next .. '(' .. i .. '): ' .. tostring(pipeline):gsub(line, line .. tab .. ext)
end
str = str .. line .. tab .. last .. 'output'
str = str .. line .. '}'

0 comments on commit 6402f6b

Please sign in to comment.
Something went wrong with that request. Please try again.