IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

ONNX CI improvement (#3673)

Browse Source 
* Improve corner-case resistance

* small improvements
This commit is contained in:
Rafal Blaczkowski 2020-12-25 05:46:55 +01:00 committed by GitHub
parent 1aee9f9ffe
commit 6d45a9c041
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
ngraph/python/tests/test_onnx/model_zoo_preprocess.sh
View File

@ -71,6 +71,8 @@ function pull_and_postprocess_onnx_model_zoo() {
mkdir -p test_data_set_0
mv *.pb test_data_set_0/
# Save SHA of successfully post processed repository
git rev-parse HEAD > onnx_sha
}
function update_onnx_models() {
@ -81,7 +83,7 @@ function update_onnx_models() {
pull_and_postprocess_onnx_model_zoo
else
# Check if ONNX Model Zoo directory consists of proper git repo
git_remote_url=`git -C $ONNX_MODELS_DIR config --local remote.origin.url 2> /dev/null 2>&1`
export git_remote_url=`git -C $ONNX_MODELS_DIR config --local remote.origin.url 2> /dev/null 2>&1`
printf "ONNX Model Zoo repository exists: %s\n" "$ONNX_MODELS_DIR"
if [[ $git_remote_url = "https://github.com/onnx/models.git" ]]; then
printf "The proper github repository detected: %s\n" "$git_remote_url"
@ -92,7 +94,7 @@ function update_onnx_models() {
fi
cd "$ONNX_MODELS_DIR"
CURRENT_ONNX_MODELS_SHA=`git rev-parse HEAD`
export CURRENT_ONNX_MODELS_SHA=`head -n1 onnx_sha 2> /dev/null 2>&1`
if [[ $ONNX_SHA = $CURRENT_ONNX_MODELS_SHA ]] ; then
echo "ONNX Model Zoo repository is in the right state"
else