From: Francois Fleuret 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)