Skip to main content

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

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. However, you can install Torch Model Archiver stand alone.

Detailed documentation and examples are provided in the README.

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

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page