Evaluation results on specific layers

2

You could register a forward hook to the specific layer and analyze the activation after the forward pass as shown in this example.

1 Like