fix fintune RMSE and memory issue

Fix #2472. The previous implementation tried to allocate a (N, N) array and the RMSE result was actually MAE instead.

Signed-off-by: Jinzhe Zeng <[email protected]>

deepmd/fit/ener.py

@@ -856,9 +856,12 @@ def change_energy_bias(
             delta_bias = np.linalg.lstsq(type_numbs, bias_diff, rcond=None)[0]
             unbias_e = energy_predict + type_numbs @ delta_bias
             atom_numbs = type_numbs.sum(-1)
-            rmse_ae = (
-                np.sqrt(np.square(unbias_e - energy_ground_truth)) / atom_numbs
-            ).mean()
+            rmse_ae = np.sqrt(
+                np.mean(
+                    np.square(unbias_e.ravel() - energy_ground_truth.ravel())
+                    / atom_numbs
+                )
+            )
             self.bias_atom_e[idx_type_map] += delta_bias.reshape(-1)
             log.info(
                 f"RMSE of atomic energy after linear regression is: {rmse_ae} eV/atom."