(self, module, input, output)
| 485 | self.captured = None |
| 486 | |
| 487 | def forward(self, module, input, output): |
| 488 | if self.capture_input: |
| 489 | self.captured = input[0].detach() |
| 490 | else: |
| 491 | self.captured = output.detach() |
| 492 | |
| 493 | |
| 494 | def get_dstore_path(dstore_dir, model_type, eval_subset, dimension): |
nothing calls this directly
no outgoing calls
no test coverage detected