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 <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!") 
\ No newline at end of file
+    print("[INFO] Finished the analysis!")