From 1bc832ac69797a2fabdb4f1dcf758cb415e4f215 Mon Sep 17 00:00:00 2001
From: Francois Fleuret <francois@fleuret.org>
Date: Tue, 6 Dec 2016 09:15:27 +0100
Subject: [PATCH] Highlight the non-containers with '*' in profiler.print() to
 ease the reading.

---
 profiler.lua | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/profiler.lua b/profiler.lua
index 4e45787..3bbccaf 100644
--- 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)
-- 
2.20.1