yamle.utils.export_utils module

yamle.utils.export_utils.export_onnx(model, path)

This method is used to export the model to ONNX.

Parameters:

• model (nn.Module) – The model to export.

• path (str) – The path to save the model.

Return type:

None