waifu2x/lib/PrintTable.lua

local PrintTable, parent = torch.class('w2nn.PrintTable','nn.Module')

function PrintTable:__init(id)
   parent.__init(self)
   self.id = id
end
function PrintTable:updateOutput(input)
   print("----", self.id)
   print(input)
   self.output = input
   return self.output
end
function PrintTable:updateGradInput(input, gradOutput)
   self.gradInput = gradOutput
   return self.gradInput
end