INNER CODE UNIT · Python

__iter__

MyNiuuu/MOFA-Video · MOFA-Video-Hybrid/sadtalker_audio2pose/src/face3d/data/__init__.py:111

    def __iter__(self):
        """Return a batch of data"""
        for i, data in enumerate(self.dataloader):
            if i * self.opt.batch_size >= self.opt.max_dataset_size:
                break
            yield data

View source record →