Hello ptrblck!
I have an additional question…
I want to save inputs (or feature space) of each modules in a dictionary wise
so… for example if I have module names as model.layer1.conv1, model.layer1.conv2
I would like to save their input in a form of
input['model.layer1.conv1'] = [input]
and append them as iteration goes
input['model.layer1.conv1'].append(input)
I would like to do it with forward hook
I’m not sure if the code you presented here does the job for me
thanks a lot