Get Started
Blog
Docs
PyTorch Lightning TorchMetrics Lightning Flash Lightning Transformers Lightning Bolts
GitHub
Train on the cloud with Lightning

Table of Contents

latest

Get started

Quick Start

Tasks

Language Modeling
Multiple Choice
Question Answering
Summarization
Text Classification
Token Classification
Translation
Image Classification

Features

SparseML
HuggingFace Hub Checkpoints
Big Transformers Model Inference
DeepSpeed Training with Big Transformer Models

Advanced

Custom Data Files
Customizing Datasets
Customizing Tasks

API Reference

Task API Reference
DataModule API Reference

Docs >
DataModule API Reference
Edit on GitHub

Shortcuts

DataModule API Reference¶

Built with Sphinx using a theme provided by Read the Docs.

DataModule API Reference

To analyze traffic and optimize your experience, we serve cookies on this site. By clicking or navigating, you agree to allow our usage of cookies. Read PyTorch Lightning's Privacy Policy.

Get Started
Blog
Ecosystem

PyTorch Lightning
TorchMetrics
Lightning Flash
Lightning Transformers
Lightning Bolts

Github
Grid.ai