https://github.com/d-michail/pytorch-lightning-demo
Simple Pytorch Lighning Demo
https://github.com/d-michail/pytorch-lightning-demo
deep-learning lightning pytorch
Last synced: 3 months ago
JSON representation
Simple Pytorch Lighning Demo
- Host: GitHub
- URL: https://github.com/d-michail/pytorch-lightning-demo
- Owner: d-michail
- License: apache-2.0
- Created: 2024-05-10T15:21:17.000Z (about 2 years ago)
- Default Branch: main
- Last Pushed: 2025-11-07T12:14:57.000Z (8 months ago)
- Last Synced: 2025-11-07T14:19:42.605Z (8 months ago)
- Topics: deep-learning, lightning, pytorch
- Language: Python
- Homepage:
- Size: 16.6 KB
- Stars: 0
- Watchers: 1
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# Pytorch Lightning Demo
This is a very simple pytorch lightning demo to bootstrap a new project. It uses the MNIST dataset
and two different models. Uses Lightning CLI. Each different model has each own config file.
Should be enough in order to bootstrap a project for a single dataset and comparison of different models.
Run using
```
python main.py fit --config config-vit.yaml
```
or
```
python main.py fit --config config-simplevit.yaml
```
In order to install dependencies
```
python3 -m venv .venv
source .venv/bin/activate
pip install -r requirements.txt
```
Enjoy!