Runs metrics and histogram summaries at epoch end.
(self, epoch, logs=None)
| 1668 | self._set_default_writer(self._train_run_name) |
| 1669 | |
| 1670 | def on_epoch_end(self, epoch, logs=None): |
| 1671 | """Runs metrics and histogram summaries at epoch end.""" |
| 1672 | self._log_metrics(logs, prefix='epoch_', step=epoch) |
| 1673 | |
| 1674 | if self.histogram_freq and epoch % self.histogram_freq == 0: |
| 1675 | self._log_weights(epoch) |
| 1676 | |
| 1677 | if self.embeddings_freq and epoch % self.embeddings_freq == 0: |
| 1678 | self._log_embeddings(epoch) |
| 1679 | |
| 1680 | def on_train_end(self, logs=None): |
| 1681 | if self._is_tracing: |