Now takes the total time estimated at the top if totalTime is not provided.

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

diff --git a/test-profiler.lua b/test-profiler.lua
index aa6e800..71127a0 100755 (executable)
--- a/test-profiler.lua
+++ b/test-profiler.lua
@@ -93,9 +93,8 @@ end
 
 -- Print the accumulated timings
 
--- profiler.color = false
-profiler.print(model, nbSamples, modelTime)
--- profiler.print(model)
+-- profiler.print(model, nbSamples)
+profiler.print(model)
 
 print(string.format('Total model time %.02fs', modelTime))
 print(string.format('Total data time %.02fs', dataTime))