write_metadata_into_onnx
-
説明:カスタムメタデータを ONNX モデルに書き込みます。
-
パラメータ
- onnx_path (
Union[str, Path]
):ONNX モデルのパス。 - out_path (
Union[str, Path]
):出力先 ONNX モデルのパス。 - drop_old_meta (
bool
):古いメタデータを削除するかどうか。デフォルトはFalse
。 **kwargs
:カスタムメタデータ。
- onnx_path (
-
例
import capybara as cb
onnx_path = 'model.onnx'
out_path = 'model_with_metadata.onnx'
cb.write_metadata_into_onnx(
onnx_path,
out_path,
drop_old_meta=False,
key1='value1',
key2='value2',
key3='value3',
)