To get intermediate activations, you could use forward hooks.
If that’s not possible for your use case, could you post the error message you are getting?
To get intermediate activations, you could use forward hooks.
If that’s not possible for your use case, could you post the error message you are getting?