Keep production startup independent of benchmarks
This commit is contained in:
@@ -7,6 +7,7 @@ from typing import Any
|
||||
|
||||
import numpy as np
|
||||
import torch
|
||||
import cv2
|
||||
|
||||
from .core import normalize_bbox, normalize_keypoints
|
||||
|
||||
@@ -45,6 +46,7 @@ class Models:
|
||||
return digest.hexdigest()
|
||||
|
||||
def load(self) -> None:
|
||||
cv2.setNumThreads(int(os.getenv("CV2_NUM_THREADS", "2")))
|
||||
torch.backends.cudnn.benchmark = True
|
||||
torch.backends.cuda.matmul.allow_tf32 = True
|
||||
torch.backends.cudnn.allow_tf32 = True
|
||||
|
||||
Reference in New Issue
Block a user