fix num threads.

praxxus11 · Aug 6, 2024 · 91911a8 · 91911a8
1 parent 0676238
commit 91911a8
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/inference/init.py b/inference/init.py
@@ -1,5 +1,6 @@
 import logging
 import os
+import torch
 
 def init():
     logging.basicConfig(
@@ -8,4 +9,5 @@ def init():
         level=logging.DEBUG,
         format='%(asctime)s.%(msecs)03d - %(name)s - %(levelname)s - %(message)s',
         datefmt='%Y-%m-%d %H:%M:%S'
-    )
+    )
+    torch.set_num_threads(4)
diff --git a/inference/storage/db.py b/inference/storage/db.py
@@ -31,7 +31,6 @@ def __init__(self, sqlite_db_path):
             FOREIGN KEY (img_id) REFERENCES Meta (img_id)
         )
         ''')
-
         self.con.commit()
 
     def insert_into_meta(self, meta):
-Original file line number
+Diff line change
@@ Expand Up / @@ -31,7 +31,6 @@ def __init__(self, sqlite_db_path): @@
                 FOREIGN KEY (img_id) REFERENCES Meta (img_id)
             )
             ''')
             self.con.commit()
         def insert_into_meta(self, meta):
@@ Expand Down @@