Skip to content

GitLab

Commit da2c777a authored by Hongkun Yu's avatar Hongkun Yu Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 355282227
parent a0eade79
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/benchmark/retinanet_benchmark.py
View file @ da2c777a
...@@ -356,7 +356,7 @@ class ShapeMaskBenchmarkReal(DetectionBenchmarkReal): ...@@ -356,7 +356,7 @@ class ShapeMaskBenchmarkReal(DetectionBenchmarkReal):
def __init__(self, **kwargs): def __init__(self, **kwargs):
super(ShapeMaskBenchmarkReal, self).__init__(model='shapemask', super(ShapeMaskBenchmarkReal, self).__init__(model='shapemask',
per_gpu_batch_size=2, per_gpu_batch_size=4,
**kwargs) **kwargs)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment