Update.

[profiler-torch.git] / test-profiler.lua

diff --git a/test-profiler.lua b/test-profiler.lua
index 2f1f0ec..6d98b2a 100755 (executable)
--- a/test-profiler.lua
+++ b/test-profiler.lua
@@ -49,6 +49,7 @@ local model = nn.Sequential()
 -- Decor it for profiling
 
 profiler.decor(model)
+print()
 
 -- Create the data and criterion
 
@@ -85,8 +86,9 @@ end
 
 -- Print the accumulated timings
 
-profiler.print(model, nbSamples)
+-- profiler.print(model, nbSamples)
+profiler.print(model)
 
-print('----------------------------------------------------------------------')
+print()
 print(string.format('Total model time %.02fs', modelTime))
 print(string.format('Total data time %.02fs', dataTime))