projects
/
profiler-torch.git
/ blobdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
|
commitdiff
|
tree
raw
|
inline
| side by side
OCD cosmetics.
[profiler-torch.git]
/
test-profiler.lua
diff --git
a/test-profiler.lua
b/test-profiler.lua
index
aa6e800
..
6d79e91
100755
(executable)
--- a/
test-profiler.lua
+++ b/
test-profiler.lua
@@
-51,12
+51,9
@@
local model = nn.Sequential()
)
:add(nn.Linear(1000, 100))
)
:add(nn.Linear(1000, 100))
--- Decor it for profiling
+-- Decor
ate
it for profiling
profiler.decorate(model)
profiler.decorate(model)
-print()
-
-torch.save('model.t7', model)
-- Create the data and criterion
-- Create the data and criterion
@@
-93,8
+90,8
@@
end
-- Print the accumulated timings
-- Print the accumulated timings
--- profiler.color = false
-profiler.print(model, nbSamples
, modelTime
)
+print()
+profiler.print(model, nbSamples)
-- profiler.print(model)
print(string.format('Total model time %.02fs', modelTime))
-- profiler.print(model)
print(string.format('Total model time %.02fs', modelTime))