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: 8 days 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 (6 months ago)
• Default Branch: main
• Last Pushed: 2024-05-28T08:01:14.000Z (6 months ago)
• Last Synced: 2024-05-28T19:15:28.702Z (6 months ago)
• Topics: deep-learning, lightning, pytorch
• Language: Python
• Homepage:
• Size: 13.7 KB
• Stars: 0
• Watchers: 2
• 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 --model ViT --config config-vit.yaml

```

or 

```

python main.py fit --model SimpleViT --config config-simplevit.yaml

```

In order to install dependencies

```

python3 -m venv .venv

source .venv/bin/activate 

pip install -r requirements.txt

```

Enjoy!