X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?p=profiler-torch.git;a=blobdiff_plain;f=test-profiler.lua;fp=test-profiler.lua;h=a78c944464b8a147894650924b34256ab8ebbba6;hp=7c22576420c80875c0204e555708fdec9a5abe6a;hb=c2a852c17eb19f472762343ee8dfe3e0d1b40dbf;hpb=77fb30ba6143770ed71e68d89bde96c82ae8f1a0

diff --git a/test-profiler.lua b/test-profiler.lua
index 7c22576..a78c944 100755
--- 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)