pstorch

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

Install

pip install pstorch

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:

python main.py

CLI

pstorch generate pytorch --input-dim 784 --output-dim 10 --hidden 256,128 --metrics accuracy,loss

Defaults example:

pstorch generate

Named default example:

pstorch generate pytorch

Output:

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