From 200c617d86032b4e6b16cb25f26453af084c9859 Mon Sep 17 00:00:00 2001 From: mckay 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(