Browse Source

style: del log

main
mckay 3 months ago
parent
commit
97e0cf9699
  1. 6
      brep2sdf/train.py

6
brep2sdf/train.py

@ -140,9 +140,9 @@ class Trainer:
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}")
#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