From 15cf564c2e3164f129699d9b4f3cf6c42376bcb9 Mon Sep 17 00:00:00 2001 From: unanmed <1319491857@qq.com> Date: Sun, 5 Apr 2026 22:16:01 +0800 Subject: [PATCH] =?UTF-8?q?chore:=20=E8=B0=83=E6=95=B4=20heatmap=20?= =?UTF-8?q?=E6=A8=A1=E5=9E=8B=E5=8F=82=E6=95=B0?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- ginka/train_heatmap.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ginka/train_heatmap.py b/ginka/train_heatmap.py index 05cdd27..06898d8 100644 --- a/ginka/train_heatmap.py +++ b/ginka/train_heatmap.py @@ -26,8 +26,8 @@ from .utils import nms_sampling # 0. 墙壁热力图, 1. 怪物热力图, 2. 资源热力图, 3. 血瓶热力图, 4. 宝石热力图, 5. 钥匙热力图 # 6. 道具热力图, 7. 入口热力图, 8. 门热力图 -BATCH_SIZE = 8 -VAL_BATCH_DIVIDER = 8 +BATCH_SIZE = 128 +VAL_BATCH_DIVIDER = 64 NUM_CLASSES = 16 MASK_TOKEN = 15 GENERATE_STEP = 8