Update.

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

diff --git a/test-profiler.lua b/test-profiler.lua
index 6d98b2a..7c22576 100755 (executable)
--- a/test-profiler.lua
+++ b/test-profiler.lua
@@ -86,8 +86,8 @@ end
 
 -- Print the accumulated timings
 
--- profiler.print(model, nbSamples)
-profiler.print(model)
+profiler.print(model, nbSamples)
+-- profiler.print(model)
 
 print()
 print(string.format('Total model time %.02fs', modelTime))