From 035394febd5963b42abf33528bbb8123e56c1b49 Mon Sep 17 00:00:00 2001
From: songhyeonsu <hunsoo0823@naver.com>
Date: Fri, 8 May 2026 16:56:09 +0900
Subject: [PATCH] Drop batch size to 16 (32 OOMs on 5090 with PGNet's variable
 input size)
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

PGNet uses dynamic image sizes per batch (max_text_size=512), which spikes
peak memory above 32GB at batch 32. Settle on 16 — ~2x throughput vs the
default 14 while staying well clear of OOM.

Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
---
 configs/kr_lp_pgnet.yml | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/configs/kr_lp_pgnet.yml b/configs/kr_lp_pgnet.yml
index baa306e..6d23c49 100644
--- a/configs/kr_lp_pgnet.yml
+++ b/configs/kr_lp_pgnet.yml
@@ -83,7 +83,7 @@ Train:
           channel_first: False
       - E2ELabelEncodeTrain:
       - PGProcessTrain:
-          batch_size: 32
+          batch_size: 16
           use_resize: True
           use_random_crop: False
           min_crop_size: 24
@@ -95,7 +95,7 @@ Train:
   loader:
     shuffle: True
     drop_last: True
-    batch_size_per_card: 32
+    batch_size_per_card: 16
     num_workers: 8
 
 Eval: