From 55c4c525faa323c4bc55f71b51449fcb27b2829a Mon Sep 17 00:00:00 2001 From: Tom Wildenhain Date: Tue, 1 Dec 2020 21:29:14 -0500 Subject: [PATCH] Added covid-twitter-bert to perf testing script Signed-off-by: Tom Wildenhain --- tests/completed_perf_testing_models.yaml | 29 ++++++++++++++++++++++++ 1 file changed, 29 insertions(+) diff --git a/tests/completed_perf_testing_models.yaml b/tests/completed_perf_testing_models.yaml index 002785f5f..b0ace156a 100644 --- a/tests/completed_perf_testing_models.yaml +++ b/tests/completed_perf_testing_models.yaml @@ -3330,3 +3330,32 @@ universal-sentence-encoder-qa: atol: 0.0005 tag: "serve" signature_def: "default" + +covid-twitter-bert: + disabled: false + skip_conversion: true + model: "C:/Users/tomwi/Documents/tfhubmodels/covid-twitter-bert" + converted_model: "C:/Users/tomwi/Documents/tfhubmodels/covid-twitter-bert/model.onnx" + model_type: saved_model + large_model: false + run_tf_frozen: false + use_custom_ops: true + input_get: get_small_rand_int32 + inputs: + "input_word_ids:0": [1, 512] + "input_mask:0": + shape: [1, 512] + input_get: get_ones_int32 + "input_type_ids:0": + shape: [1, 512] + input_get: get_zeros_then_ones + outputs: + - Identity:0 + - Identity_1:0 + structured_outputs: + - transformer_encoder + - transformer_encoder_1 + rtol: 0.05 + atol: 0.0005 + tag: "serve" + signature_def: "serving_default"