Skip to content

Commit

Permalink
added extra model/MLPerf path printing
Browse files Browse the repository at this point in the history
  • Loading branch information
gfursin committed Feb 29, 2024
1 parent 7f418eb commit 974e27c
Show file tree
Hide file tree
Showing 18 changed files with 54 additions and 2 deletions.
3 changes: 3 additions & 0 deletions cm-mlops/script/get-ml-model-3d-unet-kits19/_cm.json
Original file line number Diff line number Diff line change
Expand Up @@ -87,5 +87,8 @@
"CM_UNZIP": "yes"
}
}
},
"print_env_at_the_end" : {
"CM_ML_MODEL_FILE_WITH_PATH": "Path to the ML model"
}
}
3 changes: 3 additions & 0 deletions cm-mlops/script/get-ml-model-abtf-ssd-pytorch/_cm.json
Original file line number Diff line number Diff line change
Expand Up @@ -74,5 +74,8 @@
},
"group": "epoch"
}
},
"print_env_at_the_end" : {
"CM_ML_MODEL_FILE_WITH_PATH": "Path to the ML model"
}
}
3 changes: 3 additions & 0 deletions cm-mlops/script/get-ml-model-bert-base-squad/_cm.json
Original file line number Diff line number Diff line change
Expand Up @@ -85,5 +85,8 @@
},
"group": "precision"
}
},
"print_env_at_the_end" : {
"CM_ML_MODEL_BERT_VOCAB_FILE_WITH_PATH": "Path to the BERT vocab file"
}
}
3 changes: 3 additions & 0 deletions cm-mlops/script/get-ml-model-bert-large-squad/_cm.json
Original file line number Diff line number Diff line change
Expand Up @@ -355,5 +355,8 @@
"CM_BERT_"
]
}
},
"print_env_at_the_end" : {
"CM_ML_MODEL_FILE_WITH_PATH": "Path to the ML model"
}
}
3 changes: 3 additions & 0 deletions cm-mlops/script/get-ml-model-dlrm-terabyte/_cm.json
Original file line number Diff line number Diff line change
Expand Up @@ -157,5 +157,8 @@
"CM_DLRM_MULTIHOT_MODEL": "yes"
}
}
},
"print_env_at_the_end" : {
"CM_ML_MODEL_FILE_WITH_PATH": "Path to the ML model"
}
}
5 changes: 4 additions & 1 deletion cm-mlops/script/get-ml-model-efficientnet-lite/_cm.json
Original file line number Diff line number Diff line change
Expand Up @@ -160,5 +160,8 @@
[ "lite2", "resolution-260" ],
[ "lite3", "resolution-280" ],
[ "lite4", "resolution-300" ]
]
],
"print_env_at_the_end" : {
"CM_ML_MODEL_FILE_WITH_PATH": "Path to the ML model"
}
}
3 changes: 3 additions & 0 deletions cm-mlops/script/get-ml-model-gptj/_cm.json
Original file line number Diff line number Diff line change
Expand Up @@ -210,5 +210,8 @@
},
"pytorch,int4,intel": {
}
},
"print_env_at_the_end" : {
"CM_ML_MODEL_FILE_WITH_PATH": "Path to the ML model"
}
}
3 changes: 3 additions & 0 deletions cm-mlops/script/get-ml-model-huggingface-zoo/_cm.json
Original file line number Diff line number Diff line change
Expand Up @@ -80,6 +80,9 @@
}
]
}
},
"print_env_at_the_end" : {
"CM_ML_MODEL_FILE_WITH_PATH": "Path to the ML model"
}
}

3 changes: 3 additions & 0 deletions cm-mlops/script/get-ml-model-llama2/_cm.json
Original file line number Diff line number Diff line change
Expand Up @@ -120,5 +120,8 @@
},
"group": "precision"
}
},
"print_env_at_the_end" : {
"LLAMA2_CHECKPOINT_PATH": "LLAMA2 checkpoint path"
}
}
3 changes: 3 additions & 0 deletions cm-mlops/script/get-ml-model-mobilenet/_cm.json
Original file line number Diff line number Diff line change
Expand Up @@ -375,5 +375,8 @@
"tf"
]
}
},
"print_env_at_the_end" : {
"CM_ML_MODEL_FILE_WITH_PATH": "Path to the ML model"
}
}
3 changes: 3 additions & 0 deletions cm-mlops/script/get-ml-model-neuralmagic-zoo/_cm.json
Original file line number Diff line number Diff line change
Expand Up @@ -295,5 +295,8 @@
"CM_MODEL_ZOO_STUB": "#"
}
}
},
"print_env_at_the_end" : {
"CM_ML_MODEL_FILE_WITH_PATH": "Path to the ML model"
}
}
3 changes: 3 additions & 0 deletions cm-mlops/script/get-ml-model-resnet50/_cm.json
Original file line number Diff line number Diff line change
Expand Up @@ -300,5 +300,8 @@
},
"group": "precision"
}
},
"print_env_at_the_end" : {
"CM_ML_MODEL_FILE_WITH_PATH": "Path to the ML model"
}
}
3 changes: 3 additions & 0 deletions cm-mlops/script/get-ml-model-retinanet-nvidia/_cm.json
Original file line number Diff line number Diff line change
Expand Up @@ -77,5 +77,8 @@
"CM_NVIDIA_EFFICIENT_NMS": "yes"
}
}
},
"print_env_at_the_end" : {
"CM_NVIDIA_RETINANET_EFFICIENT_NMS_CONCAT_MODEL_WITH_PATH": "Path to the ML model"
}
}
2 changes: 1 addition & 1 deletion cm-mlops/script/get-ml-model-retinanet/_cm.json
Original file line number Diff line number Diff line change
Expand Up @@ -166,6 +166,6 @@
}
},
"print_env_at_the_end" : {
"CM_ML_MODEL_FILE_WITH_PATH": "Path to the model"
"CM_ML_MODEL_FILE_WITH_PATH": "Path to the ML model"
}
}
3 changes: 3 additions & 0 deletions cm-mlops/script/get-ml-model-rnnt/_cm.json
Original file line number Diff line number Diff line change
Expand Up @@ -67,5 +67,8 @@
"group": "download-src",
"default": true
}
},
"print_env_at_the_end" : {
"CM_ML_MODEL_FILE_WITH_PATH": "Path to the ML model"
}
}
3 changes: 3 additions & 0 deletions cm-mlops/script/get-ml-model-stable-diffusion/_cm.json
Original file line number Diff line number Diff line change
Expand Up @@ -175,5 +175,8 @@
"CM_DOWNLOAD_URL": "mlc-inference:mlcommons-inference-wg-public/stable_diffusion_fp16"
}
}
},
"print_env_at_the_end" : {
"SDXL_CHECKPOINT_PATH": "Stable diffusion checkpoint path"
}
}
3 changes: 3 additions & 0 deletions cm-mlops/script/get-ml-model-tiny-resnet/_cm.json
Original file line number Diff line number Diff line change
Expand Up @@ -136,5 +136,8 @@
}
}
}
},
"print_env_at_the_end" : {
"CM_ML_MODEL_FILE_WITH_PATH": "Path to the ML model"
}
}
4 changes: 4 additions & 0 deletions cm-mlops/script/get-mlperf-inference-src/_cm.json
Original file line number Diff line number Diff line change
Expand Up @@ -230,5 +230,9 @@
"CM_GIT_CHECKOUT": "pybind_update"
}
}
},
"print_env_at_the_end" : {
"CM_MLPERF_INFERENCE_SOURCE": "Path to MLPerf inference benchmark sources",
"CM_MLPERF_INFERENCE_CONF_PATH": "Path to the MLPerf inference benchmark configuration file"
}
}

0 comments on commit 974e27c

Please sign in to comment.