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
Functions
get_eval_dataloader
get_train_dataloader
dict containing input parameters for creating dataset.
previous
modelzoo.fc_mnist.pytorch
next
modelzoo.fc_mnist.pytorch.data.get_eval_dataloader