You saw the lightning. Now it's time to hear the thunder 🌩️

Thunder 🌩️

The Deep Learning framework based on Lightning.

Install

pip install thunder

Start experimenting

It's as simple as 1, 2, 3:

1. Create a config (e.g. base.config):

   from myproject import MyDataset, MyModule
   from lightning import Trainer
   from torch.utils.data import DataLoader
   
   # these 3 fields are required
   train_data = DataLoader(MyDataset())
   module = MyModule()
   trainer = Trainer()

2. Build the experiment:

   thunder build base.config /path/to/some/folder

3. Run it

   thunder run /path/to/some/folder

Also, 2 and 3 can be combined into a single command:

thunder build-run base.config /path/to/some/folder

More advanced stuff

See our docs for a full list of neat things thunder🌩️ can do for you