jih332:

Hey thanks for the reply. I’ve thought about that too but if it was done for the purpose of reducing computation, shouldn’t it be

backward hook in module layer4/2/relu
backward hook in module layer4/2/bn3
backward hook in module layer4/2/conv3



You are right. Since it was backward I read the prints from the bottom to the top and got confused