def METHOD_NAME(self, use_tqdm=False):
    """Manages the (reloading) data adapter before epoch starts."""
    for epoch in tqdm.tqdm(
        range(self._initial_epoch, self._epochs), disable=not use_tqdm
    ):
        self._adapter.reload()
        yield epoch, iter(self._adapter.get_tf_generator())
        self._adapter.on_epoch_end()
    self._adapter.on_complete()