Browse Source

feat: SDF值分布log

main
mckay 3 months ago
parent
commit
200c617d86
  1. 4
      brep2sdf/train.py

4
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(

Loading…
Cancel
Save