modlee.converter

The converter module holds the Converter class for converting between different formats of a neural network. The supported formats include Torch and ONNX.

The Torch formats include:

• Model, the object which contains the forward pass and can be trained

• Code, the model as text code that can be saved to a file and used to rebuild the model

The ONNX formats include:

• Graph, the network represented as a graph with layers as nodes

• Text, the textual description of the graph that is portable and can be rebuilt into a graph

Classes