Highlight the non-containers with '*' in profiler.print() to ease the reading.

diff --git a/profiler.lua b/profiler.lua
index 4e45787..3bbccaf 100644 (file)
--- a/profiler.lua
+++ b/profiler.lua
@@ -82,8 +82,15 @@ end
 
 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)