Removed the signature.

[profiler-torch.git] / profiler.lua

diff --git a/profiler.lua b/profiler.lua
index 231a7c9..b363a60 100644 (file)
--- a/profiler.lua
+++ b/profiler.lua
@@ -55,7 +55,7 @@ function profiler.decorate(model, functionsToDecorate)
                   .. tostring(model))
       end
 
-      local toDecorate = model
+      local toDecorate = getmetatable(model)
 
       while not rawget(toDecorate, name) do
          toDecorate = getmetatable(toDecorate)
@@ -123,15 +123,14 @@ function profiler.print(model, nbSamples, totalTime, indent)
       hint = ' '
    else
       if profiler.color then
-         hint = ' '
+         hint = '  ' .. profiler.colors('red')
       else
-         hint = '*'
+         hint = '* '
       end
-      hint = hint .. profiler.colors('red')
    end
 
-   print(profiler.timingString(indent .. hint .. ' ' .. model.__typename,
-                         localTotal, nbSamples, totalTime))
+   print(profiler.timingString(indent .. hint .. model.__typename,
+                               localTotal, nbSamples, totalTime))
 
    for l, t in pairs(model.accTime) do
       print(profiler.timingString(indent .. '  :' .. l, t, nbSamples, totalTime))