function profiler.print(model, nbSamples, totalTime, indent)
local indent = indent or ''
+ local hint
- print(string.format('%s* %s', indent, model.__typename))
+ if torch.isTypeOf(model, nn.Container) then
+ hint = ' '
+ else
+ hint = '*'
+ end
+
+ print(string.format('%s%s %s', indent, hint, model.__typename))
for l, t in pairs(model.accTime) do
local s = string.format('%s %s %.02fs', indent, l, t)