From f49da7c255ca6bf97388d984e8a91a8f5e62f239 Mon Sep 17 00:00:00 2001 From: mckay Date: Fri, 9 May 2025 19:33:10 +0800 Subject: [PATCH] =?UTF-8?q?=E6=A2=AF=E5=BA=A6=E8=AE=A1=E7=AE=97=20?= =?UTF-8?q?=E5=AE=89=E5=85=A8=E5=BD=92=E4=B8=80=E5=8C=96=20=EF=BC=8C=20?= =?UTF-8?q?=E6=A2=AF=E5=BA=A6=E5=8F=AF=E8=83=BD=E4=BC=9A=E5=BE=88=E5=B0=8F?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- brep2sdf/networks/network.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/brep2sdf/networks/network.py b/brep2sdf/networks/network.py index 033e07a..60d9782 100644 --- a/brep2sdf/networks/network.py +++ b/brep2sdf/networks/network.py @@ -223,9 +223,9 @@ def gradient(inputs, outputs): # 修正维度切片方式 if points_grad is None: return torch.zeros_like(inputs[:, -3:]) # 处理空梯度情况 - + #logger.debug(f"points_grad:{points_grad},shape:{points_grad.shape}") # 添加安全截取和归一化 coord_grad = points_grad[:, -3:] if points_grad.shape[1] >=3 else points_grad - coord_grad = coord_grad / (coord_grad.norm(dim=-1, keepdim=True) + 1e-6) # 安全归一化 + coord_grad = coord_grad / (coord_grad.norm(dim=-1, keepdim=True) + 1e-16) # 安全归一化 return coord_grad \ No newline at end of file