X-Git-Url: https://www.fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=blobdiff_plain;f=test-profiler.lua;h=6d98b2a22a9c6badb4ce4e3a48dda9f69d7041d5;hb=c4b8d97c31f5e7dc948bf51ad6564d24e377801b;hp=2f1f0ec7883d993eb222ba5f7404da1257e7e41d;hpb=1cc41a13ae500ceaf037b3ff880a0eeb25472633;p=profiler-torch.git

diff --git a/test-profiler.lua b/test-profiler.lua
index 2f1f0ec..6d98b2a 100755
--- 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))