Introduction
Get Started
Prepare your data
Cerebras PyTorch API
cerebras_pytorch
cerebras_pytorch.amp
cerebras_pytorch.optim
cerebras_pytorch.sparse
cerebras_pytorch.metrics
Cerebras Model Zoo
Cerebras Guides
Fundamentals
Support
Bases: object
object
Methods
create_dataloader
get_augment_transforms
get_tile_transform
preprocess_image
preprocess_mask
transform_image_and_mask
Attributes
tiling_image_shape
previous
modelzoo.vision.pytorch.unet.input.UNetDataProcessor.visualize_dataset
next
modelzoo.vision.pytorch.unet.input.preprocessing_utils