Skip to content

Commit

Permalink
Documantation for artifact format (#212)
Browse files Browse the repository at this point in the history
Shubhada-Deshmukh authored Aug 18, 2022
1 parent ede0dc0 commit 0aa60de
Showing 1 changed file with 21 additions and 0 deletions.
21 changes: 21 additions & 0 deletions docs/artifact-format.md
Original file line number Diff line number Diff line change
@@ -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" | - |

0 comments on commit 0aa60de

Please sign in to comment.