Method batch_generator

ppo/ppo_datahelper.py:353–359

(self)

        return batch
            
    def batch_generator(self):
        while True:
            for batch in super().batch_generator():
                if len(batch) == self.batch_size:
                    yield batch
            if self.mode != 'train':
                break