Pytorch-Lightning-template

  • Pytorch Lightning Modularize boilerplate code
python3 -m venv venv
source venv/bin/activate
pip install torch torchvision torchmetrics python_lightning numpy==1.26.0
# change data directory in config.py 
python3  main.py

Output

prediction-out