From c5fb66e1dc32db33d76d56c48740869066ec875c Mon Sep 17 00:00:00 2001
From: Simon Fan <xmfan@meta.com>
Date: Mon, 23 Oct 2023 20:18:42 +0000
Subject: [PATCH] yolov3: reduce batch size due to OOM

---
 torchbenchmark/models/yolov3/__init__.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/torchbenchmark/models/yolov3/__init__.py b/torchbenchmark/models/yolov3/__init__.py
index 2abdf9307f..14037cec10 100644
--- a/torchbenchmark/models/yolov3/__init__.py
+++ b/torchbenchmark/models/yolov3/__init__.py
@@ -30,7 +30,7 @@ class Model(BenchmarkModel):
     task = COMPUTER_VISION.SEGMENTATION
     # Original train batch size: 16
     # Source: https://github.com/ultralytics/yolov3/blob/master/train.py#L447
-    DEFAULT_TRAIN_BSIZE = 16
+    DEFAULT_TRAIN_BSIZE = 8 # reduced from 16 to 8 due to OOM in TorchInductor dashboard
     DEFAULT_EVAL_BSIZE = 8
     # yolov3 CUDA inference test uses amp precision
     DEFAULT_EVAL_CUDA_PRECISION = "amp"