fix(pt): wrap fparam/aparam reshape with descriptive ValueError

Match the dpmodel try/except pattern so shape mismatches produce
a clear error instead of a raw RuntimeError from torch.view.

Author: Han Wang

deepmd/pt/model/task/fitting.py

@@ -779,7 +779,13 @@ def _forward_common(
             assert fparam is not None, "fparam should not be None"
             assert self.fparam_avg is not None
             assert self.fparam_inv_std is not None
-            fparam = fparam.view([nf, self.numb_fparam])
+            try:
+                fparam = fparam.view([nf, self.numb_fparam])
+            except RuntimeError as e:
+                raise ValueError(
+                    f"input fparam: cannot reshape {list(fparam.shape)} "
+                    f"into ({nf}, {self.numb_fparam})."
+                ) from e
             nb, _ = fparam.shape
             t_fparam_avg = self._extend_f_avg_std(self.fparam_avg, nb)
             t_fparam_inv_std = self._extend_f_avg_std(self.fparam_inv_std, nb)
@@ -799,7 +805,13 @@ def _forward_common(
             assert aparam is not None, "aparam should not be None"
             assert self.aparam_avg is not None
             assert self.aparam_inv_std is not None
-            aparam = aparam.view([nf, -1, self.numb_aparam])
+            try:
+                aparam = aparam.view([nf, -1, self.numb_aparam])
+            except RuntimeError as e:
+                raise ValueError(
+                    f"input aparam: cannot reshape {list(aparam.shape)} "
+                    f"into ({nf}, nloc, {self.numb_aparam})."
+                ) from e
             nb, nloc, _ = aparam.shape
             t_aparam_avg = self._extend_a_avg_std(self.aparam_avg, nb, nloc)
             t_aparam_inv_std = self._extend_a_avg_std(self.aparam_inv_std, nb, nloc)