dont set threads using torch

BIMSBbioinfo · Jul 18, 2024 · 21303c9 · 21303c9
1 parent ad0309d
commit 21303c9
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/flexynesis/__main__.py b/flexynesis/__main__.py
@@ -141,12 +141,10 @@ def main():
                                    "`srun --gpus=1 --pty flexynesis <rest of your_command>` !!!\n\n"]))
             time.sleep(3)  #wait a bit to capture user's attention to the warning
             device_type = 'cpu'
-            torch.set_num_threads(args.threads)
         else:
             device_type = 'gpu'
     else:
         device_type = 'cpu'
-        torch.set_num_threads(args.threads)
 
     # 5. check GNN arguments
     if args.model_class == 'GNN':
@@ -407,4 +405,4 @@ def main():
 
 if __name__ == "__main__":
     main()
-    print("[INFO] Finished the analysis!") 
+    print("[INFO] Finished the analysis!")