(self, writer_name)
| 1617 | } |
| 1618 | |
| 1619 | def _increment_step(self, writer_name): |
| 1620 | step = self._total_batches_seen[writer_name] |
| 1621 | if isinstance(step, variables.Variable): |
| 1622 | step.assign_add(1) |
| 1623 | else: |
| 1624 | self._total_batches_seen[writer_name] += 1 |
| 1625 | |
| 1626 | def on_train_begin(self, logs=None): |
| 1627 | self._init_batch_steps() |
no test coverage detected