pycasx.tools.nn_load

Function to load all neural network models from ONNX files.

Module Contents

Functions

load_nnet_from_path(nnet_path)	Load all neural network models from NNet files.
load_onnx_from_path(onnx_path)	Load all neural network models from ONNX files.
load_torch_from_path(onnx_path)	Load all neural network models from ONNX files.

pycasx.tools.nn_load.load_nnet_from_path(nnet_path)

Load all neural network models from NNet files.

Parameters:

nnet_path (Path | str) – Path to the NNet files.

Returns:

Dictionary of neural network models.

Return type:

dict[str, NNet]

pycasx.tools.nn_load.load_onnx_from_path(onnx_path)

Load all neural network models from ONNX files.

Parameters:

onnx_path (Path | str) – Path to the ONNX files.

Returns:

Dictionary of neural network models.

Return type:

dict[str, ModelProto]

pycasx.tools.nn_load.load_torch_from_path(onnx_path)

Load all neural network models from ONNX files.

Parameters:

onnx_path (Path | str) – Path to the ONNX files.

Returns:

Dictionary of neural network models.

Return type:

dict[str, GraphModule]