(self, layer, capture_input=False)
| 478 | |
| 479 | class ActivationCapturer(nn.Module): |
| 480 | def __init__(self, layer, capture_input=False): |
| 481 | super().__init__() |
| 482 | self.layer = layer |
| 483 | self.capture_input = capture_input |
| 484 | |
| 485 | self.captured = None |
| 486 | |
| 487 | def forward(self, module, input, output): |
| 488 | if self.capture_input: |