From 200c617d86032b4e6b16cb25f26453af084c9859 Mon Sep 17 00:00:00 2001
From: mckay <wchpub@163.com>
Date: Tue, 3 Dec 2024 21:20:55 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20SDF=E5=80=BC=E5=88=86=E5=B8=83log?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 brep2sdf/train.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/brep2sdf/train.py b/brep2sdf/train.py
index ceaccf2..7f5063b 100644
--- a/brep2sdf/train.py
+++ b/brep2sdf/train.py
@@ -139,6 +139,10 @@ class Trainer:
                 edge_mask=edge_mask,
                 query_points=points
             )
+
+            logger.info("\n=== SDF值分布 ===")
+            logger.info(f"Pred SDF - min: {pred_sdf.min().item():.6f}, max: {pred_sdf.max().item():.6f}, mean: {pred_sdf.mean().item():.6f}")
+            logger.info(f"GT SDF - min: {gt_sdf.min().item():.6f}, max: {gt_sdf.max().item():.6f}, mean: {gt_sdf.mean().item():.6f}")
             
             # 计算损失
             loss = self.criterion(