(self)
| 262 | self._writer.add_graph(self.graph) |
| 263 | |
| 264 | def _before_train(self): |
| 265 | # Writing the graph is expensive (takes ~2min) when the graph is large. |
| 266 | # Therefore use a separate thread. It will then run in the |
| 267 | # background while TF is warming up in the first several iterations. |
| 268 | self._write_graph_thread = threading.Thread(target=self._write_graph) |
| 269 | self._write_graph_thread.daemon = True |
| 270 | self._write_graph_thread.start() |
| 271 | |
| 272 | @HIDE_DOC |
| 273 | def process_summary(self, summary): |