Add SpatialReSamplingEx (grouping all the Spatial*Sampling).

SpatialPadding can now pad on dimensions different than (2,3).
Add Tic/Toc modules, to time a network

1 files changed, 19 insertions, 0 deletions

diff --git a/Toc.lua b/Toc.lua
new file mode 100644
index 0000000..059211e
--- /dev/null
+++ b/Toc.lua
@@ -0,0 +1,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