Update.
[profiler-torch.git] / test-profiler.lua
index c43995d..6d79e91 100755 (executable)
@@ -55,10 +55,6 @@ local model = nn.Sequential()
 
 profiler.decorate(model)
 
-print()
-
--- torch.save('model.t7', model)
-
 -- Create the data and criterion
 
 local input = torch.Tensor(1000, 1, h, w)
@@ -94,8 +90,9 @@ end
 
 -- Print the accumulated timings
 
--- profiler.print(model, nbSamples)
-profiler.print(model)
+print()
+profiler.print(model, nbSamples)
+-- profiler.print(model)
 
 print(string.format('Total model time %.02fs', modelTime))
 print(string.format('Total data time %.02fs', dataTime))