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=c43995d9ab0640a06d1fe43e8aea278b8cc5078c;hp=71127a0a7965e03b1c47c5d5ec16f5ee19f628cb;hb=b758f38fa8142256d754a5ba18d6a7932d6e203c;hpb=3debf6e5063b03dc620d2ed67c6c155245bfedd8 diff --git a/test-profiler.lua b/test-profiler.lua index 71127a0..c43995d 100755 --- a/test-profiler.lua +++ b/test-profiler.lua @@ -51,12 +51,13 @@ local model = nn.Sequential() ) :add(nn.Linear(1000, 100)) --- Decor it for profiling +-- Decorate it for profiling profiler.decorate(model) + print() -torch.save('model.t7', model) +-- torch.save('model.t7', model) -- Create the data and criterion