Toc.lua


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19

require 'nn'
local Toc, parent = torch.class('nn.Toc', 'nn.Module')

function Toc:__init(name, comment)
   parent.__init(self)
   self.name = name or 'default'
   self.comment = comment or ''
end

function Toc:updateOutput(input)
   print("Toc '"..self.name.."' ("..self.comment..") : "..tic_modules[self.name]:time()['real'])
   self.output = input
   return self.output
end

function Toc:updateGradInput(input, gradOutput) 
   self.gradInput = gradOutput
   return self.gradInput
end