Hi,
Why is the forward pre-hook necessary here? Why not include it at the beginning of the forward?
Or if you want to inherit the forward from the parent class, create a new forward, do the preprocessing and then call the parent forward with super().forward(args).