onnx

lanfactory.onnx.transform_onnx

transform_to_onnx

transform_to_onnx(network_config_file, state_dict_file, input_shape, output_onnx_file)

Transforms a TorchMLP model to ONNX format.

Arguments

network_config_file (str):
    Path to the pickle file containing the network configuration.
state_dict_file (str):
    Path to the file containing the state dictionary of the model.
input_shape (int):
    The size of the input tensor for the model.
output_onnx_file (str):
    Path to the output ONNX file.

lanfactory.onnx.transform_to_onnx

transform_to_onnx(network_config_file, state_dict_file, input_shape, output_onnx_file)

Transforms a TorchMLP model to ONNX format.

Arguments

network_config_file (str):
    Path to the pickle file containing the network configuration.
state_dict_file (str):
    Path to the file containing the state dictionary of the model.
input_shape (int):
    The size of the input tensor for the model.
output_onnx_file (str):
    Path to the output ONNX file.