I don’t think you need hooks to see the output of each layer, you can just log it in the forward function,
For example you can just print it
x1 = self.fc_spinal_layer1(x[:, 0:Half_width])
print(x1, x1.shape)

Or maybe I am misunderstanding something?