Skip to content

Latest commit

 

History

History
14 lines (11 loc) · 359 Bytes

File metadata and controls

14 lines (11 loc) · 359 Bytes

Cookiecutter template for basic packaging of PyTorch models

pip install cookiecutter
cookiecutter https://github.com/ml-illustrated/cookiecutter-pytorch-basic
# fill in your project info

cd <name_of_your_project>
# add your model definition into <name_of_your_project>/model.py, etc.
# ...

# build your python package
python setup.py bdist_wheel