Getting Started
Concepts
Model Zoo
CS Torch
cerebras.pytorch
cerebras.pytorch.amp
cerebras.pytorch.optim
cerebras.pytorch.sparse
cerebras.pytorch.metrics
Cluster Monitoring
Fundamentals
Support
Classes
MaskTransform
data[mask < 0] = 0 Sets everything outside the mask to 0.
previous
cerebras.modelzoo.data.vision.segmentation.transforms.crop_and_pad_augmentations.random_crop
next
cerebras.modelzoo.data.vision.segmentation.transforms.custom_transforms.MaskTransform