fix: dtype

deepmodeling · Mar 26, 2024 · daa73b9 · daa73b9
1 parent 858eed4
commit daa73b9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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: