Skip to content

Commit 24c7868

Browse files
committed
Update run_pretrained_models.yaml
1 parent 5798722 commit 24c7868

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

tests/run_pretrained_models.yaml

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -122,10 +122,15 @@ esrgan-tf2:
122122
model: ersgan
123123
model_type: saved_model
124124
input_get: get_beach
125+
opset_constraints:
126+
"onnx":
127+
"min": 10
125128
inputs:
126129
"input_0:0": [1, 50, 50, 3]
127130
outputs:
128131
- Identity:0
132+
rtol: 0.02
133+
atol: 0.0005
129134

130135
inception_v3_slim:
131136
url: https://storage.googleapis.com/download.tensorflow.org/models/inception_v3_2016_08_28_frozen.pb.tar.gz

0 commit comments

Comments
 (0)