From 0aa60def64a18d21e1778ef910e428f06f0819c5 Mon Sep 17 00:00:00 2001 From: sdeshmuk1 <108807773+sdeshmuk1@users.noreply.github.com> Date: Thu, 18 Aug 2022 04:21:02 +0200 Subject: [PATCH] Documantation for artifact format (#212) --- docs/artifact-format.md | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) create mode 100644 docs/artifact-format.md diff --git a/docs/artifact-format.md b/docs/artifact-format.md new file mode 100644 index 0000000..b3f520d --- /dev/null +++ b/docs/artifact-format.md @@ -0,0 +1,21 @@ +# Preparing Artifact format + + When saving an artifact with ORMB you have to declare its format in ormbfile.yaml, which is case sensitive. Here is the list of the accepted format values and extensions. + +| Format | Required format value | Required artifact extension | +| ------------- | ------------- | ------------- | +| SavedModel | "SavedModel" | "saved_model.pb" | +| ONNX | "ONNX" | ".onnx" | +| H5 | "H5" | ".h5" | +| PMML | "PMML" | ".pmml" | +| CaffeModel | "CaffeModel" | ".caffemodel" | +| NetDef | "NetDef" | "init_net.pb" | +| MXNetParams | "MXNetParams" | ".params" | +| TorchScript | "TorchScript" | ".pt" | +| GraphDef | "GraphDef" | ".graphdef" | +| TensorRT | "TensorRT" | ".engine" | +| SKLearn | "SKLearn" | ".joblib" | +| XGBoost | "XGBoost" | ".xgboost" | +| MLflow | "MLflow" | - | +| Pickle | "Pickle" | ".pickle" | +| Others | "Others" | - |