From: Francois Fleuret Date: Mon, 5 Dec 2016 16:25:38 +0000 (+0100) Subject: Tidied a bit the printing. X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=commitdiff_plain;h=40c92fc3324ed00f30d0f01f6b6c4010b2a1e1fe;p=profiler-torch.git Tidied a bit the printing. --- diff --git a/profiler.lua b/profiler.lua index b95d750..ad280d2 100644 --- a/profiler.lua +++ b/profiler.lua @@ -76,21 +76,14 @@ function profiler.print(model, nbSamples, indent) print(string.format('%s* %s', indent, model.__typename)) - if nbSamples then - for l, t in pairs(model.accTime) do - print(string.format('%s %s %.02fs (%.01fmus/sample)', - indent, - l, - t, - 1e6 * t / nbSamples)) - end - else - for l, t in pairs(model.accTime) do - print(string.format('%s %s %.02fs', - indent, - l, - t)) + for l, t in pairs(model.accTime) do + local s + if nbSamples then + s = string.format(' (%.01fmus/sample)', 1e6 * t / nbSamples) + else + s = '' end + print(string.format('%s %s %.02fs%s', indent, l, t, s)) end if torch.isTypeOf(model, nn.Container) then