Limit the number of models run on OpenVINO (#3742)

* Removed NMS from supported list
This commit is contained in:
suryasidd 2020-04-29 02:23:09 -07:00 committed by GitHub
parent 7ff06056bd
commit e529464a12
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 3 deletions

View file

@ -17,7 +17,7 @@ sess = onnxruntime.InferenceSession(<path_to_model_file>, options)
2. C++ API
```
SessionOptions::SetGraphOptimizationLevel(ORT_DISABLE_ALL);
SessionOptions::SetGraphOptimizationLevel(ORT_DISABLE_ALL);
```
## Dynamic device selection
@ -71,7 +71,6 @@ VPUs as well as Intel<sup>®</sup> Vision accelerator Design with Intel Movidiu
| MaxPool | Yes | Yes | Yes |
| Min | Yes | Yes | Yes |
| Mul | Yes | Yes | Yes |
| NonMaxSuppression | Yes | Yes | Yes |
| Pad | Yes | Yes | Yes |
| Pow | Yes | Yes | Yes |
| PRelu | Yes | Yes | Yes |

View file

@ -1266,7 +1266,7 @@ def openvino_run_onnx_tests(build_dir, configs, onnx_test_data_dir,
model_dir_opset8, "test_*"))
model_dir_opset10 = os.path.join(model_dir, "opset10")
model_dir_opset10 = glob.glob(os.path.join(
model_dir_opset10, "tf_*"))
model_dir_opset10, "*v1*"))
for dir_path in itertools.chain(model_dir_opset8,
model_dir_opset10):
model_test_cmd = cmd + [dir_path]