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

.. py:module:: pycasx.tools.nn_load

.. autoapi-nested-parse::

   Function to load all neural network models from ONNX files.



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


Functions
~~~~~~~~~

.. autoapisummary::

   pycasx.tools.nn_load.load_nnet_from_path
   pycasx.tools.nn_load.load_onnx_from_path
   pycasx.tools.nn_load.load_torch_from_path



.. py:function:: load_nnet_from_path(nnet_path)

   Load all neural network models from NNet files.

   :param nnet_path: Path to the NNet files.
   :type nnet_path: Path | str

   :returns: Dictionary of neural network models.
   :rtype: dict[str, NNet]


.. py:function:: load_onnx_from_path(onnx_path)

   Load all neural network models from ONNX files.

   :param onnx_path: Path to the ONNX files.
   :type onnx_path: Path | str

   :returns: Dictionary of neural network models.
   :rtype: dict[str, ModelProto]


.. py:function:: load_torch_from_path(onnx_path)

   Load all neural network models from ONNX files.

   :param onnx_path: Path to the ONNX files.
   :type onnx_path: Path | str

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