fix also pd

Author: Han Wang

deepmd/pd/loss/ener.py

@@ -211,8 +211,8 @@ def forward(self, input_dict, model, label, natoms, learning_rate, mae=False):
                     loss += atom_norm * (pref_e * l2_ener_loss)
                 else:
                     l_huber_loss = custom_huber_loss(
-                        atom_norm * model_pred["energy"],
-                        atom_norm * label["energy"],
+                        atom_norm * energy_pred,
+                        atom_norm * energy_label,
                         delta=self.huber_delta,
                     )
                     loss += pref_e * l_huber_loss