waifu2x/lib/Print.lua

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

function Print:__init(id)
   parent.__init(self)
   self.id = id
end
function Print:updateOutput(input)
   print("----", self.id)
   print(input:size())
   self.output:resizeAs(input)
   self.output:copy(input)
   return self.output
end
function Print:updateGradInput(input, gradOutput)
   self.gradInput:resizeAs(GradOutput)
   return self.gradInput
end