From 3d53de09d8c1fb904d5cb0a137b9bc75e9d77108 Mon Sep 17 00:00:00 2001 From: chflame163 Date: Sun, 11 Jan 2026 12:52:41 +0800 Subject: [PATCH] fix debug info --- py/data_nodes.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/py/data_nodes.py b/py/data_nodes.py index 036543d..c25c414 100644 --- a/py/data_nodes.py +++ b/py/data_nodes.py @@ -554,7 +554,7 @@ class LS_MultiImageListToBatch: # 转回 [B, H, W, C] batch = batch.permute(0, 2, 3, 1) - print(f"batch.shape={batch.shape}") + sizes.append(batch.shape[0]) out.append(batch)