Skip to content

Commit 55c4c52

Browse files
Added covid-twitter-bert to perf testing script
Signed-off-by: Tom Wildenhain <[email protected]>
1 parent f1bb5e1 commit 55c4c52

File tree

1 file changed

+29
-0
lines changed

1 file changed

+29
-0
lines changed

tests/completed_perf_testing_models.yaml

Lines changed: 29 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3330,3 +3330,32 @@ universal-sentence-encoder-qa:
33303330
atol: 0.0005
33313331
tag: "serve"
33323332
signature_def: "default"
3333+
3334+
covid-twitter-bert:
3335+
disabled: false
3336+
skip_conversion: true
3337+
model: "C:/Users/tomwi/Documents/tfhubmodels/covid-twitter-bert"
3338+
converted_model: "C:/Users/tomwi/Documents/tfhubmodels/covid-twitter-bert/model.onnx"
3339+
model_type: saved_model
3340+
large_model: false
3341+
run_tf_frozen: false
3342+
use_custom_ops: true
3343+
input_get: get_small_rand_int32
3344+
inputs:
3345+
"input_word_ids:0": [1, 512]
3346+
"input_mask:0":
3347+
shape: [1, 512]
3348+
input_get: get_ones_int32
3349+
"input_type_ids:0":
3350+
shape: [1, 512]
3351+
input_get: get_zeros_then_ones
3352+
outputs:
3353+
- Identity:0
3354+
- Identity_1:0
3355+
structured_outputs:
3356+
- transformer_encoder
3357+
- transformer_encoder_1
3358+
rtol: 0.05
3359+
atol: 0.0005
3360+
tag: "serve"
3361+
signature_def: "serving_default"

0 commit comments

Comments
 (0)