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