Update deepmd/dpmodel/atomic_model/polar_atomic_model.py

Signed-off-by: Jinzhe Zeng <[email protected]>
njzjz · Nov 20, 2024 · c4c98b2 · c4c98b2
1 parent 5f41c02
commit c4c98b2
Showing 1 changed file with 0 additions and 1 deletion.
diff --git 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,