Add back Keras Mask RCNN to CUDA test exclusion (#2201)

This commit is contained in:
Hariharan Seshadri 2019-10-20 17:13:02 -07:00 committed by GitHub
parent aae18a3fe3
commit 07e9f500da
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -521,6 +521,7 @@ int real_main(int argc, char* argv[], Ort::Env& env) {
#endif
#ifdef USE_CUDA
broken_tests.insert({"mask_rcnn_keras", "result mismatch"});
broken_tests.insert({"mlperf_ssd_mobilenet_300", "unknown error"});
broken_tests.insert({"mlperf_ssd_resnet34_1200", "unknown error"});
broken_tests.insert({"tf_inception_v1", "flaky test"}); //TODO: Investigate cause for flakiness