INNER CODE UNIT · Python

__len__

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

    def __len__(self):
        """Return the number of data in the dataset"""
        return min(len(self.dataset), self.opt.max_dataset_size)

    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 →