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(