Could you try the following:
# Create the model
model = models.resnet152(pretrained=True)
# Load the state_dict
checkpoint = torch.load('model_best.pth.tar')
model.load_state_dict(checkpoint['state_dict'])
model.eval()

After that try to register the hooks for your desired layer.