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=ed6a1cb695e566af881fbaa7024fc34d98e5393d;hp=c43995d9ab0640a06d1fe43e8aea278b8cc5078c;hb=9e12d0e90f4c03891fb1fa1f61870b92930f26ad;hpb=b758f38fa8142256d754a5ba18d6a7932d6e203c diff --git a/test-profiler.lua b/test-profiler.lua index c43995d..ed6a1cb 100755 --- a/test-profiler.lua +++ b/test-profiler.lua @@ -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))