Skip to content

Commit

Permalink
added vgg11 model to regression test suite
Browse files Browse the repository at this point in the history
  • Loading branch information
harshbafna committed Jul 28, 2020
1 parent 7961914 commit 5ffd0d7
Showing 1 changed file with 27 additions and 0 deletions.
27 changes: 27 additions & 0 deletions test/postman/inference_data.json
Original file line number Diff line number Diff line change
Expand Up @@ -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",
Expand Down

0 comments on commit 5ffd0d7

Please sign in to comment.