    def on_train_batch_begin(self, batch: Tuple, logs: Dict) -> None:
        """
        Hook function invoked before train stage.

        Args:
            logs (`Dict`):
               Kv store to save and load info.
        """
        self.on_batch_begin(batch, logs)
Hook function invoked before train stage.

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