AncTe/ginka-generator
1
0
Fork 0
mirror of https://github.com/unanmed/ginka-generator.git synced 2026-05-17 23:21:20 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

chore: 调整 heatmap 模型参数

Browse Source
This commit is contained in:
unanmed 2026-04-05 22:16:01 +08:00
parent b4f49e702d
commit 15cf564c2e
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
ginka/train_heatmap.py
View File

@ -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