https://github.com/stackloklabs/pytorch-pipeline
Example pipeline to explore model provenance
https://github.com/stackloklabs/pytorch-pipeline
Last synced: 3 months ago
JSON representation
Example pipeline to explore model provenance
- Host: GitHub
- URL: https://github.com/stackloklabs/pytorch-pipeline
- Owner: StacklokLabs
- Created: 2024-02-22T09:28:59.000Z (about 1 year ago)
- Default Branch: main
- Last Pushed: 2024-02-25T15:23:00.000Z (about 1 year ago)
- Last Synced: 2024-02-25T16:31:56.829Z (about 1 year ago)
- Language: Python
- Size: 21.5 KB
- Stars: 0
- Watchers: 2
- Forks: 1
- Open Issues: 1
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
# Simple PyTorch Model Pipeline
This is a simple PyTorch model pipeline that can be used to train and evaluate a model. The pipeline is designed to be used with a simple feedforward neural network, but can be easily adapted to other models.
This is for experimental around model provenance and integrity.
A simple github action is used to run the pipeline.
## Usage
Set up a virtualenv and install the requirements:
```bash
virtualenv -p python3 venv
source venv/bin/activate
pip install -r requirements.txt
```
Generate the training set:
```bash
python generate_dataset.py
```
Train the model:
```bash
python train_model.py
```
Inference:
```bash
python run_inference.py
```