(self, len_dataset: int)
| 21 | self._reset(len_dataset) |
| 22 | |
| 23 | def eval(self, len_dataset: int) -> None: |
| 24 | self.flush() |
| 25 | self.is_train = False |
| 26 | self._reset(len_dataset) |
| 27 | |
| 28 | def __call__(self, model, loss, accuracy, learning_rate: float = None) -> None: |
| 29 | if self.is_train: |