(self, metadata)
| 228 | show_dataflow=True, show_memory=True)) |
| 229 | |
| 230 | def _write_event(self, metadata): |
| 231 | evt = tf.Event() |
| 232 | evt.tagged_run_metadata.tag = 'trace-{}'.format(self.global_step) |
| 233 | evt.tagged_run_metadata.run_metadata = metadata.SerializeToString() |
| 234 | self.trainer.monitors.put_event(evt) |
| 235 | |
| 236 | |
| 237 | class GPUMemoryTracker(Callback): |
no test coverage detected