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: torch.utils.data.Dataset
torch.utils.data.Dataset
Methods
collate
Call self as a function.
previous
modelzoo.transformers.pytorch.layers_api_demo.data.train_input_dataloader
next
modelzoo.transformers.pytorch.layers_api_demo.model