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

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

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

Methods

check_split_valid

create_dataloader

create_dataset

custom_collate_fn

process_transform