diff --git a/deepmd/pt/utils/utils.py b/deepmd/pt/utils/utils.py
index 95ffaafd50..ac251648af 100644
--- a/deepmd/pt/utils/utils.py
+++ b/deepmd/pt/utils/utils.py
@@ -118,7 +118,7 @@ def dict_to_device(sample_dict):
             sample_dict[key] = [item.to(DEVICE) for item in sample_dict[key]]
         if isinstance(sample_dict[key], np.float32):
             sample_dict[key] = (
-                torch.ones(1, dtype=torch.float64, device=DEVICE) * sample_dict[key]
+                torch.ones(1, dtype=torch.float32, device=DEVICE) * sample_dict[key]
             )
         else:
             if sample_dict[key] is not None: