    def on_epoch_end(self, epochs: int, logs: Dict) -> None:
        """
        Hook function invoked after each epoch.

        Args:
            epochs (`int`):
                Epoch index.
            logs (`Dict`):
                Kv store to save and load info.
        """
        pass
Hook function invoked after each epoch.

Args:
epochs (`int`):
Epoch index.
logs (`Dict`):
Kv store to save and load info.
towhee/trainer/callback.py