I assume “features” refers to an intermediate forward activation. If so, you could use forward hooks as described here to grab them and that you could store them as numpy arrays afterwards.
1 Like
I assume “features” refers to an intermediate forward activation. If so, you could use forward hooks as described here to grab them and that you could store them as numpy arrays afterwards.