Is there a way to stop the model’s forward function early using forward hooks without modifying the original forward function?
No, I don’t think you can use forward hooks to stop the model execution.
1 Like