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