Provide a scope for running one epoch.
(self, epoch=0, mode=ModeKeys.TRAIN)
| 686 | |
| 687 | @tf_contextlib.contextmanager |
| 688 | def on_epoch(self, epoch=0, mode=ModeKeys.TRAIN): |
| 689 | """Provide a scope for running one epoch.""" |
| 690 | epoch_logs = {} |
| 691 | if mode == ModeKeys.TRAIN: |
| 692 | self.callbacks.on_epoch_begin(epoch, epoch_logs) |
| 693 | self.progbar.on_epoch_begin(epoch, epoch_logs) |
| 694 | try: |
| 695 | yield epoch_logs |
| 696 | finally: |
| 697 | if mode == ModeKeys.TRAIN: |
| 698 | # Epochs only apply to `fit`. |
| 699 | self.callbacks.on_epoch_end(epoch, epoch_logs) |
| 700 | self.progbar.on_epoch_end(epoch, epoch_logs) |
| 701 | |
| 702 | @tf_contextlib.contextmanager |
| 703 | def on_batch(self, step=0, mode=ModeKeys.TRAIN, size=1): |
no test coverage detected