(self, _, rv)
| 278 | return None |
| 279 | |
| 280 | def _after_run(self, _, rv): |
| 281 | results = rv.results |
| 282 | if results is not None: |
| 283 | for mem, dev in zip(results, self._devices): |
| 284 | self.trainer.monitors.put_scalar('PeakMemory(MB)' + dev, mem / 1e6) |
| 285 | |
| 286 | |
| 287 | PeakMemoryTracker = GPUMemoryTracker |
nothing calls this directly
no test coverage detected