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
Bases: object
object
Saves the value of data_dict[in_key] to data_dict[out_key]. Optionally removes data_dict[in_key] from the dict.
Methods
previous
cerebras.modelzoo.data.vision.segmentation.transforms.utility_transforms.RemoveLabelTransform
next
cerebras.modelzoo.data.vision.segmentation.transforms.utils