From 8328afa882803f0cb7714f303fb441b657ce1a8d Mon Sep 17 00:00:00 2001
From: Evgeniya Egupova <evgeniya.egupova@intel.com>
Date: Thu, 25 Mar 2021 12:33:20 +0300
Subject: [PATCH] Limit the batch size of retinanet in sota checkpoint test
 (#597)

---
 beta/tests/sota_checkpoints_eval.json | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/beta/tests/sota_checkpoints_eval.json b/beta/tests/sota_checkpoints_eval.json
index 2d599bfd0ef..ec2e4a0a29a 100644
--- a/beta/tests/sota_checkpoints_eval.json
+++ b/beta/tests/sota_checkpoints_eval.json
@@ -149,7 +149,8 @@
                     "target": 33.44,
                     "weights": "retinanet/retinanet.h5",
                     "metric_type": "mAP",
-                    "model_description": "retinanet"
+                    "model_description": "retinanet",
+                    "batch": 40
                 },
                 "retinanet_int8_w_sym_t_a_sym_t": {
                     "config": "examples/tensorflow/object_detection/configs/quantization/retinanet_coco_int8.json",
@@ -158,7 +159,8 @@
                     "resume": "retinanet_int8_w_sym_t_a_sym_t",
                     "metric_type": "mAP",
                     "model_description": "retinanet_int8_w_sym_t_a_sym_t",
-                    "compression_description": "INT8"
+                    "compression_description": "INT8",
+                    "batch": 40
                 },
                 "retinanet_sparsity_50": {
                     "config": "examples/tensorflow/object_detection/configs/sparsity/retinanet_coco_magnitude_sparsity.json",
@@ -167,7 +169,8 @@
                     "resume": "retinanet_sparsity_50",
                     "metric_type": "mAP",
                     "model_description": "retinanet_sparsity_50",
-                    "compression_description": "Sparsity 50"
+                    "compression_description": "Sparsity 50",
+                    "batch": 40
                 }
             }
         }