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
training_data_generator
Generator function used to create input dataset for GPT2Model.
previous
modelzoo.transformers.pytorch.gpt2.input.scripts
next
modelzoo.transformers.pytorch.gpt2.input.scripts.data_processor_utils.training_data_generator