Skip to content

Commit 3d1efe5

Browse files
author
wayuanho
authored
Merge pull request #645 from zhijxu-MS/enable_two_model_tests
Enable two model tests
2 parents 6eae86e + 1f28eca commit 3d1efe5

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tests/run_pretrained_models.yaml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -239,7 +239,6 @@ resnet50_v2_nchw: # NOTE: Tensorflow 1.9.0 fails
239239

240240

241241
resnet50_v2_nhwc:
242-
disabled: true # FIXME: numerical results are correct but shape=(64,) for tf and (64,1) for lotus
243242
model_type: saved_model
244243
url: http://download.tensorflow.org/models/official/20181001_resnet/savedmodels/resnet_v2_fp32_savedmodel_NHWC.tar.gz
245244
model: resnet_v2_fp32_savedmodel_NHWC/1538687283
@@ -291,10 +290,11 @@ style-transfer:
291290
- output:0
292291

293292
ssd_mobilenet_v1_coco:
294-
# issues with control flow
295-
disabled: true
296293
url: http://download.tensorflow.org/models/object_detection/ssd_mobilenet_v1_coco_2017_11_17.tar.gz
297294
model: ssd_mobilenet_v1_coco_2017_11_17/frozen_inference_graph.pb
295+
opset_constraints:
296+
"onnx":
297+
"min": 10
298298
input_get: get_beach
299299
inputs:
300300
"image_tensor:0": [1, 224, 224, 3]

0 commit comments

Comments
 (0)