From 21303c9e3d60a430947dcdbbd7e8464c5d1e92fb Mon Sep 17 00:00:00 2001 From: Bora Uyar Date: Thu, 18 Jul 2024 13:47:37 +0200 Subject: [PATCH] dont set threads using torch --- flexynesis/__main__.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/flexynesis/__main__.py b/flexynesis/__main__.py index 4509207..e37cd9b 100644 --- a/flexynesis/__main__.py +++ b/flexynesis/__main__.py @@ -141,12 +141,10 @@ def main(): "`srun --gpus=1 --pty flexynesis ` !!!\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!") \ No newline at end of file + print("[INFO] Finished the analysis!")