cerebras.modelzoo.data.vision.diffusion.DiffusionLatentImageNet1KProcessor.CategoricalDataset#
- class cerebras.modelzoo.data.vision.diffusion.DiffusionLatentImageNet1KProcessor.CategoricalDataset(*args, **kwargs)[source]#
Bases:
torch.utils.data.Dataset
Methods