Metadata-Version: 2.4
Name: pstorch
Version: 0.1.6
Summary: The goal of this project is to make pytorch easier to use.
Description-Content-Type: text/markdown
Requires-Dist: httpx>=0.28.1
Requires-Dist: huggingface>=0.0.1
Requires-Dist: jinja2>=3.1
Requires-Dist: mlflow>=3.7.0
Requires-Dist: pydantic>=2.12.5
Requires-Dist: torch>=2.1
Requires-Dist: transformers>=4.57.3

# pstorch

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

## Install

```bash
pip install pstorch
```

## Python

```python
import pstorch

pstorch.generate(
    name="pytorch",
    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 generate pytorch --input-dim 784 --output-dim 10 --hidden 256,128 --metrics accuracy,loss
```

Defaults example:

```bash
pstorch generate
```

Named default example:

```bash
pstorch generate pytorch
```

Output:

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