diff --git a/userbenchmark/dynamo/dynamobench/torchbench.py b/userbenchmark/dynamo/dynamobench/torchbench.py index 0518b537a0..c97f829b43 100755 --- a/userbenchmark/dynamo/dynamobench/torchbench.py +++ b/userbenchmark/dynamo/dynamobench/torchbench.py @@ -417,8 +417,8 @@ def load_model( # comparison hard with torch.compile. torch.compile can cause minor # divergences in the output because of how fusion works for amp in # TorchInductor compared to eager. Therefore, instead of looking at - # all the bounding boxes, we compare only top 5. - model_kwargs = {"box_detections_per_img": 5} + # all the bounding boxes, we compare only top 4. + model_kwargs = {"box_detections_per_img": 4} benchmark = benchmark_cls( test="train", device=device,