Update.
authorFrancois Fleuret <francois@fleuret.org>
Tue, 6 Dec 2016 14:03:33 +0000 (15:03 +0100)
committerFrancois Fleuret <francois@fleuret.org>
Tue, 6 Dec 2016 14:03:33 +0000 (15:03 +0100)
test-profiler.lua

index c43995d..ed6a1cb 100755 (executable)
@@ -55,8 +55,6 @@ local model = nn.Sequential()
 
 profiler.decorate(model)
 
-print()
-
 -- torch.save('model.t7', model)
 
 -- Create the data and criterion
@@ -94,8 +92,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))