Fixed a stupid and embarrassing typo.

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

diff --git a/test-profiler.lua b/test-profiler.lua
index 6d98b2a..a78c944 100755 (executable)
--- a/test-profiler.lua
+++ b/test-profiler.lua
@@ -48,9 +48,11 @@ local model = nn.Sequential()
 
 -- Decor it for profiling
 
-profiler.decor(model)
+profiler.decorate(model)
 print()
 
+torch.save('model.t7', model)
+
 -- Create the data and criterion
 
 local input = torch.Tensor(1000, 1000)
@@ -86,8 +88,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))