Back to Bentoml

PyTorch

docs/source/reference/bentoml/frameworks/pytorch.rst

1.4.38512 B
Original Source

======= PyTorch

.. admonition:: About this page

This is an API reference for PyTorch in BentoML. Please refer to :doc:PyTorch guide </frameworks/pytorch> for more information about how to use PyTorch in BentoML.

.. note::

You can find more examples for PyTorch in our :examples:BentoML/examples <> directory.

.. currentmodule:: bentoml.pytorch

.. autofunction:: bentoml.pytorch.save_model

.. autofunction:: bentoml.pytorch.load_model

.. autofunction:: bentoml.pytorch.get