From c4c98b2c2459219478d030245b4f445bf864c741 Mon Sep 17 00:00:00 2001 From: Jinzhe Zeng Date: Tue, 19 Nov 2024 21:09:44 -0500 Subject: [PATCH] Update deepmd/dpmodel/atomic_model/polar_atomic_model.py Signed-off-by: Jinzhe Zeng --- deepmd/dpmodel/atomic_model/polar_atomic_model.py | 1 - 1 file changed, 1 deletion(-) diff --git a/deepmd/dpmodel/atomic_model/polar_atomic_model.py b/deepmd/dpmodel/atomic_model/polar_atomic_model.py index 6057c5ba65..bc7860491c 100644 --- a/deepmd/dpmodel/atomic_model/polar_atomic_model.py +++ b/deepmd/dpmodel/atomic_model/polar_atomic_model.py @@ -43,7 +43,6 @@ def apply_out_stat( dtype = out_bias[self.bias_keys[0]].dtype for kk in self.bias_keys: ntypes = out_bias[kk].shape[0] - temp = xp.zeros(ntypes, dtype=dtype) temp = xp.mean( xp.diagonal(out_bias[kk].reshape(ntypes, 3, 3), axis1=1, axis2=2), axis=1,