minor fix

keysongkang · Jun 19, 2024 · 424a07e · 424a07e
1 parent 0f706a4
commit 424a07e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/libs/lib_dft.py b/libs/lib_dft.py
@@ -10,6 +10,8 @@
 
 from libs.lib_util   import check_mkdir, single_print
 
+import torch
+torch.set_default_dtype(torch.float64)
 
 def run_DFT(inputs):
     """Function [get_criteria_uncert]