cerebras.modelzoo.data.vision.diffusion.DiffusionImageNet1KProcessor.DiffusionImageNet1KProcessor#

class cerebras.modelzoo.data.vision.diffusion.DiffusionImageNet1KProcessor.DiffusionImageNet1KProcessor(config)[source]#

Bases: cerebras.modelzoo.data.vision.diffusion.DiffusionBaseProcessor.DiffusionBaseProcessor

Methods

check_split_valid

create_dataloader

create_dataset

custom_collate_fn

process_transform

previous

cerebras.modelzoo.data.vision.diffusion.DiffusionImageNet1KProcessor

next

cerebras.modelzoo.data.vision.diffusion.DiffusionLatentImageNet1KProcessor