From 877ff5d2e28f51f8ba8b4180795396657c36bb67 Mon Sep 17 00:00:00 2001 From: Jinzhe Zeng Date: Mon, 5 Jan 2026 23:07:12 +0800 Subject: [PATCH] fix(dpmodel): rmse_e normalized by n_atoms Consistent with other backends. --- deepmd/dpmodel/loss/ener.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deepmd/dpmodel/loss/ener.py b/deepmd/dpmodel/loss/ener.py index 7ac81d8a3d..9441c97e32 100644 --- a/deepmd/dpmodel/loss/ener.py +++ b/deepmd/dpmodel/loss/ener.py @@ -180,7 +180,7 @@ def call( ) loss += pref_e * l_huber_loss more_loss["rmse_e"] = self.display_if_exist( - xp.sqrt(l2_ener_loss), find_energy + xp.sqrt(l2_ener_loss) * atom_norm_ener, find_energy ) if self.has_f: l2_force_loss = xp.mean(xp.square(diff_f))