Hi,
Be careful that the legacy module follows exactly the torch7 semantic and most of it’s api.
The regular pytorch module do not, nn.Module should be called with their input, and the function that should be implemented is forward, not updateOutput. Moreover, no updateGradInput nor updateParameters is needed because this is done by the autograd engine automatically.