# pstorch

Generate a ready-to-run PyTorch project in seconds.

## Install

```bash
pip install pstorch
```

## Python

```python
import pstorch

pstorch.generate(
    name="MyClassifier",
    arch="mlp",
    input_dim=784,
    output_dim=10,
    hidden=[256, 128],
    metrics=["accuracy", "loss"],
    prod=False,
)
```

Then run the boilerplate:

```bash
python main.py
```

## CLI

```bash
pstorch MyClassifier --arch mlp --input-dim 784 --output-dim 10 --hidden 256,128 --metrics accuracy,loss
```

Output:

```
src/
    __init__.py
    config.py
    models.py
    data.py
    train.py
    predict.py
main.py
requirements.txt
```
