Stopping forward with forward hooks

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