Fix: Square atom_norm in non-Huber energy and virial loss calculations.#5332

anyangml wants to merge 3 commits intodeepmodeling:masterdeepmodeling/deepmd-kit:masterfrom

anyangml:fix/rmse-loss-normalizationanyangml/deepmd-kit:fix/rmse-loss-normalizationCopy head branch name to clipboard