Skip to content

Latest commit

 

History

History
 
 

build_your_own_trainer

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
README.md
README.md
 
 
run.py
run.py
 
 
trainer.py
trainer.py
 
 

README.md

Build Your Own Trainer (BYOT)

This example demonstrates how easy it is to build a fully customizable trainer for your LightningModule using Fabric. It is built upon lightning.fabric for hardware and training orchestration and consists of two files:

  • trainer.py contains the actual MyCustomTrainer implementation
  • run.py contains a script utilizing this trainer for training a very simple MNIST module.

Run

To run this example, call python run.py

Requirements

This example has the following requirements which need to be installed on your python environment:

  • lightning
  • torchmetrics
  • torch
  • torchvision
  • tqdm

to install them with the appropriate versions run:

pip install "lightning>=2.0" "torchmetrics>=0.11" "torchvision>=0.14" "torch>=1.13" tqdm