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))