An open API service indexing awesome lists of open source software.

https://github.com/cr21/pytorch-lightning-template


https://github.com/cr21/pytorch-lightning-template

Last synced: 2 months ago
JSON representation

Awesome Lists containing this project

README

        

# Pytorch-Lightning-template

- Pytorch Lightning Modularize boilerplate code

```bash
python3 -m venv venv
source venv/bin/activate
pip install torch torchvision torchmetrics python_lightning numpy==1.26.0
```

```bash
# change data directory in config.py
python3 main.py
```

## Output
![prediction-out](https://i.imgur.com/g2YtB7X.png)