I have a question about “register_forward_hook”. Part of my code is as follow,
def hook(module, input, output):
It does not give me the intermediate feature. I am new to pytorch and I am not sure what is the problem.