:py:mod:`pycasx.tools.nn_convert`
=================================

.. py:module:: pycasx.tools.nn_convert

.. autoapi-nested-parse::

   Functions to convert different neural network formats to each other.



Module Contents
---------------


Functions
~~~~~~~~~

.. autoapisummary::

   pycasx.tools.nn_convert.onnx_to_torch



.. py:function:: onnx_to_torch(nets)

   Convert ONNX models to PyTorch models.

   :param nets: Dictionary of ONNX models.
   :type nets: dict[str, ModelProto]

   :returns: Dictionary of PyTorch models.
   :rtype: dict[str, GraphModule]