Solved by sakimarquis in post #7 
              
              
                Also, from one of your past answer, I found that I didn’t remove the hooks, which essencially caused the problems. 
total_feat_out = []

def hook_fn_forward(module, input, output):
    total_feat_out.append(output) 
    print(output.shape)

modules = model_ft.named_children()
handles = {}
for name, …