Fix: Invar_fitting warning msg (#3485)

This should fix #3476 --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
deepmodeling · Mar 18, 2024 · 9b6042b · 9b6042b
1 parent eca5b30
commit 9b6042b
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/deepmd/pt/model/task/invar_fitting.py b/deepmd/pt/model/task/invar_fitting.py
@@ -167,11 +167,7 @@ def compute_output_stats(
         bias_atom_e = compute_output_stats(
             merged, self.ntypes, stat_file_path, self.rcond, self.atom_ener
         )
-        self.bias_atom_e.copy_(
-            torch.tensor(bias_atom_e, device=env.DEVICE).view(
-                [self.ntypes, self.dim_out]
-            )
-        )
+        self.bias_atom_e.copy_(bias_atom_e.view([self.ntypes, self.dim_out]))
 
     def output_def(self) -> FittingOutputDef:
         return FittingOutputDef(