+--[[
+
+ Written by Francois Fleuret (francois@fleuret.org)
+
+ This is free and unencumbered software released into the public
+ domain.
+
+ Anyone is free to copy, modify, publish, use, compile, sell, or
+ distribute this software, either in source code form or as a
+ compiled binary, for any purpose, commercial or non-commercial, and
+ by any means.
+
+ In jurisdictions that recognize copyright laws, the author or
+ authors of this software dedicate any and all copyright interest in
+ the software to the public domain. We make this dedication for the
+ benefit of the public at large and to the detriment of our heirs
+ and successors. We intend this dedication to be an overt act of
+ relinquishment in perpetuity of all present and future rights to
+ this software under copyright law.
+
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+ EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+ MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+ NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS BE LIABLE FOR ANY
+ CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF
+ CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+ WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+
+ For more information, please refer to <http://unlicense.org/>
+
+]]--
+
require 'torch'
require 'nn'
require 'sys'
}
for _, name in pairs(functionsToDecorate) do
- model.timings = 0
+ model.accTime = 0
local functionTable = model
local startTime = sys.clock()
local result = { self.orig[name](self, unpack({...})) }
local endTime = sys.clock()
- self.timings = self.timings + endTime - startTime
+ self.accTime = self.accTime + endTime - startTime
return unpack(result)
end
end
print('----------------------------------------------------------------------')
print(model)
if nbSamples then
- print(string.format('acc_time %.02fs (%.1ems/sample)', model.timings, 1000 * model.timings / nbSamples))
+ print(string.format('acc_time %.02fs (%.01fmus/sample)',
+ model.accTime,
+ 1e6 * model.accTime / nbSamples))
else
- print(string.format('acc_time %.02fs', model.timings))
+ print(string.format('acc_time %.02fs', model.accTime))
end
if torch.isTypeOf(model, nn.Container) then