Can I calculate loss using intermediate feature maps obtained by forward hook?

1

Hi, I wonder if loss will backward successfully when using feature maps obtained by forward hook.

2

Yes, that should be possible as long as you don’t detach the output activations.

1 Like