Finalize batch processing

def on_batch_end(self):
        """ Finalize batch processing """
        for callback in self.callbacks:
            callback.on_batch_end(self)

        # Even with all the experience replay, we count the single rollout as a single batch
        self.epoch_info.result_accumulator.calculate(self)