def __init__(self,
                 batch_index_or_keys: Dict[str, Union[int, str]] = None,
                 outputs_index_or_keys: Dict[str, Union[int, str]] = None):
        """
        :param batch_index_or_keys: use key to support dict and index (int) to support tuple.
        :param outputs_index_or_keys: use key to support dict and index (int) to support tuple.
        """
        self.batch_index_or_keys = batch_index_or_keys
        self.outputs_index_or_keys = outputs_index_or_keys
        logger.info("Logs details:")
        logger.info(self.batch_index_or_keys)
        logger.info(self.outputs_index_or_keys)

        self.logs = {}
        self.logs_accumulation = {}

        :param batch_index_or_keys: use key to support dict and index (int) to support tuple.
        :param outputs_index_or_keys: use key to support dict and index (int) to support tuple.
        
PFPO/general_util/tensorboard_helper.py