From 5ffd0d760d48959bf1c82349df6c65f4b5b1b96f Mon Sep 17 00:00:00 2001 From: harshbafna Date: Tue, 28 Jul 2020 22:07:03 +0530 Subject: [PATCH] added vgg11 model to regression test suite --- test/postman/inference_data.json | 27 +++++++++++++++++++++++++++ 1 file changed, 27 insertions(+) diff --git a/test/postman/inference_data.json b/test/postman/inference_data.json index 9fedfaf3a9..72bd197ffb 100644 --- a/test/postman/inference_data.json +++ b/test/postman/inference_data.json @@ -167,6 +167,33 @@ }, "tolerance":1 }, + { + "url":"https://torchserve.s3.amazonaws.com/mar_files/vgg11_v2.mar", + "model_name":"vgg11", + "worker":1, + "synchronous":"true", + "file":"../examples/image_classifier/kitten.jpg", + "content-type":"application/json", + "validator":"image_classification", + "expected":[ + { + "tiger_cat":0.5501413345336914 + }, + { + "tabby":0.3231860101222992 + }, + { + "Egyptian_cat":0.11440037935972214 + }, + { + "lynx":0.006006305105984211 + }, + { + "tiger":0.004394155461341143 + } + ], + "tolerance":1 + }, { "url":"https://torchserve.s3.amazonaws.com/mar_files/vgg11_scripted.mar", "model_name":"vgg11_scripted",