Internal change

PiperOrigin-RevId: 336717785

Internal change
PiperOrigin-RevId: 336717785
cd4b8c23 · A. Unique TensorFlower · 64755c7f · cd4b8c23
Commit cd4b8c23 authored Oct 12, 2020 by A. Unique TensorFlower
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

official/benchmark/retinanet_benchmark.py official/benchmark/retinanet_benchmark.py +1 -1

No files found.
--- a/official/benchmark/retinanet_benchmark.py
+++ b/official/benchmark/retinanet_benchmark.py
@@ -356,7 +356,7 @@ class ShapeMaskBenchmarkReal(DetectionBenchmarkReal):

  def __init__(self, **kwargs):
    super(ShapeMaskBenchmarkReal, self).__init__(model='shapemask',
-                                                 per_gpu_batch_size=4,
+                                                 per_gpu_batch_size=2,
                                                 **kwargs)