Project: torch-model-archiver

Torch Model Archiver is used for creating archives of trained neural net models that can be consumed by TorchServe inference

Project Details

Latest version
0.9.0
Home Page
https://github.com/pytorch/serve/blob/master/model-archiver
PyPI Page
https://pypi.org/project/torch-model-archiver/

Project Popularity

PageRank
0.0021240867692730605
Number of downloads
250083

Project Description

Torch Model Archiver is a tool used for creating archives of trained neural net models that can be consumed by TorchServe for inference.

Use the Torch Model Archiver CLI to create a .mar file.

Torch Model Archiver is part of TorchServe <https://pypi.org/project/torchserve/>__. However, you can install Torch Model Archiver stand alone.

Detailed documentation and examples are provided in the README <https://github.com/pytorch/serve/blob/master/model-archiver/README.md>__.

Installation

::

python -m pip install torch-model-archiver

Source code

You can check the latest source code as follows:

::

git clone https://github.com/pytorch/serve.git

Citation

If you use torchserve in a publication or project, please cite torchserve: https://github.com/pytorch/serve