diff --git a/deepmd/pt/loss/ener_bind.py b/deepmd/pt/loss/ener_bind.py index d38b270b8d..334e4f2e12 100644 --- a/deepmd/pt/loss/ener_bind.py +++ b/deepmd/pt/loss/ener_bind.py @@ -404,7 +404,7 @@ def get_loss_single(input_dict_, natoms_, label_): ) loss_all += atom_norm * (pref_bind * l2_bind_loss) rmse_e = l2_bind_loss.sqrt() * atom_norm - more_loss_dict["rmse_bind"] = self.display_if_exist( + more_loss_dict["rmse_e_bind"] = self.display_if_exist( rmse_e.detach(), find_bind ) return model_pred_dict, loss_all, more_loss_dict