def METHOD_NAME(self, epoch):
    if self.is_train_split and self.do_chunk:
        old = self.chunk_size
        self.chunk_size = self.chunk_init + epoch * self.chunk_incr
        if self.chunk_bound > 0:
            self.chunk_size = min(self.chunk_size, self.chunk_bound)
        logger.info((
            f"{self.split}: setting chunk size "
            f"from {old} to {self.chunk_size}"
        ))