From: Francois Fleuret <francois@fleuret.org>
Date: Tue, 6 Dec 2016 08:15:27 +0000 (+0100)
Subject: Highlight the non-containers with '*' in profiler.print() to ease the reading.
X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=commitdiff_plain;h=1bc832ac69797a2fabdb4f1dcf758cb415e4f215;p=profiler-torch.git

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

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)