(self, logs=None)
| 1678 | self._log_embeddings(epoch) |
| 1679 | |
| 1680 | def on_train_end(self, logs=None): |
| 1681 | if self._is_tracing: |
| 1682 | self._log_trace() |
| 1683 | self._close_writers() |
| 1684 | |
| 1685 | context.context().summary_writer = self._prev_summary_writer |
| 1686 | context.context().summary_recording = self._prev_summary_recording |
| 1687 | context.context().summary_step = self._prev_summary_step |
| 1688 | |
| 1689 | def _enable_trace(self): |
| 1690 | if context.executing_eagerly(): |