This commit is contained in:
Refound-445 2024-11-16 20:52:07 +08:00
parent 26389a66ee
commit 36a06db93a

View File

@ -20,7 +20,7 @@ from ...frame_source import FrameSource
T = TypeVar("T")
device = torch.device("cuda" if config.nailong_onnx_try_to_use_gpu and torch.cuda.is_available() else "cpu")
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
@dataclass