pstorch 0.1.3

The goal of this project is to make pytorch easier to use.

pstorch

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

Install

pip install pstorch

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:

python main.py

CLI

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