`pycasx.tools.nn_convert`

Functions to convert different neural network formats to each other.

Module Contents

Functions

onnx_to_torch(nets)

Convert ONNX models to PyTorch models.

pycasx.tools.nn_convert.onnx_to_torch(nets)

Convert ONNX models to PyTorch models.

Parameters:: nets (dict[str, ModelProto]) – Dictionary of ONNX models.
Returns:: Dictionary of PyTorch models.
Return type:: dict[str, GraphModule]