From 5c1747f2264f8dd99db6531b6ce9210704ff3892 Mon Sep 17 00:00:00 2001
From: heweidong7 <511650494@qq.com>
Date: Mon, 4 Nov 2024 11:42:48 +0800
Subject: [PATCH] replace sum with 1-norm

---
 debug/accuracy_tools/kj600/kj600/features.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/debug/accuracy_tools/kj600/kj600/features.py b/debug/accuracy_tools/kj600/kj600/features.py
index 302ab3f8c5..d8ed4521b9 100644
--- a/debug/accuracy_tools/kj600/kj600/features.py
+++ b/debug/accuracy_tools/kj600/kj600/features.py
@@ -80,6 +80,7 @@ def lambda_max_subsample(module: torch.nn.Module, x: torch.tensor, num_iteration
 def cal_histc(tensor_cal, bins_total, min_val, max_val):
     return torch.histc(tensor_cal, bins=bins_total, min=min_val, max=max_val)
 
+
 @torch.no_grad()
 def get_nans(t):
-    return torch.isnan(t).sum()
+    return torch.isnan(t).float().norm(p=1)
-- 
Gitee