Limit the batch size of retinanet in sota checkpoint test (openvinoto…

…olkit#597)
evgeniya-egupova · Mar 25, 2021 · 8328afa · 8328afa
1 parent 4f0a2be
commit 8328afa
Showing 1 changed file with 6 additions and 3 deletions.
diff --git 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
                 }
             }
         }