Skip to content
Research Project Template
Text
Initializing search
Www
Research Project Template
Www
Home
Intro
Features 🔥
Features 🔥
Magic Config Schemas
Jax and Torch support with Lightning ⚡
Launching Jobs on Remote Clusters
Thorough automated testing on SLURM clusters
Examples 🧪
Examples 🧪
Image Classification (⚡)
Image Classification (jax+⚡)
NLP (🤗+⚡)
RL (jax)
Running sweeps
Profiling your code📎
Reference 🤓
Reference 🤓
Project
Project
Conftest
Experiment
Main
Main test
Algorithms
Algorithms
Example
Example test
Hf example
Hf example test
Jax example
Jax example test
Jax rl example
Jax rl example test
No op
Callbacks
Callbacks
Callback
Classification metrics
Samples per second
Testsuites
Testsuites
Algorithm tests
Configs
Configs
Config
Config test
Algorithm
Algorithm
Lr scheduler
Network
Optimizer
Datamodule
Datamodules
Datamodules
Datamodules test
Vision
Image classification
Image classification
Cifar10
Fashion mnist
Image classification
Imagenet
Imagenet32
Imagenet32 test
Inaturalist
Inaturalist test
Mnist
Text
Text
Hf text
Hf text test
Networks
Networks
Fcnet
Trainers
Trainers
Jax trainer
Utils
Utils
Autoref plugin
Autoref plugin test
Env vars
Hydra config utils
Hydra config utils test
Hydra utils
Remote launcher plugin
Remote launcher plugin test
Seeding
Testutils
Utils
Typing utils
Typing utils
Jax typing utils
Protocols
Learning Resources
Getting Help
Contributing
Text
HFDataModule
#
Bases:
LightningDataModule
Lightning data module for HF datasets.