X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=blobdiff_plain;f=test-profiler.lua;h=ed6a1cb695e566af881fbaa7024fc34d98e5393d;hb=9e12d0e90f4c03891fb1fa1f61870b92930f26ad;hp=6d98b2a22a9c6badb4ce4e3a48dda9f69d7041d5;hpb=ac1c4d6fc4926dadc6b69fd17a54419e403530a6;p=profiler-torch.git

diff --git a/test-profiler.lua b/test-profiler.lua
index 6d98b2a..ed6a1cb 100755
--- a/test-profiler.lua
+++ b/test-profiler.lua
@@ -39,21 +39,27 @@ require 'profiler'
 
 -- Create a model
 
+local w, h, fs = 50, 50, 3
+local nhu =  (w - fs + 1) * (h - fs + 1)
+
 local model = nn.Sequential()
    :add(nn.Sequential()
-           :add(nn.Linear(1000, 1000))
+           :add(nn.SpatialConvolution(1, 1, fs, fs))
+           :add(nn.Reshape(nhu))
+           :add(nn.Linear(nhu, 1000))
            :add(nn.ReLU())
        )
    :add(nn.Linear(1000, 100))
 
--- Decor it for profiling
+-- Decorate it for profiling
 
-profiler.decor(model)
-print()
+profiler.decorate(model)
+
+-- torch.save('model.t7', model)
 
 -- Create the data and criterion
 
-local input = torch.Tensor(1000, 1000)
+local input = torch.Tensor(1000, 1, h, w)
 local target = torch.Tensor(input:size(1), 100)
 local criterion = nn.MSECriterion()
 
@@ -86,9 +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))