Function save_onnx

monai/bundle/scripts.py:1421–1422

(onnx_obj: Any, filename_prefix_or_stream: str, **kwargs: Any)

    converter_kwargs_.update({"inputs": inputs_, "use_trace": use_trace_})

    def save_onnx(onnx_obj: Any, filename_prefix_or_stream: str, **kwargs: Any) -> None:
        onnx.save(onnx_obj, filename_prefix_or_stream)

    _export(
        convert_to_onnx,