Skip to content

Pytorch backend is incorrect #5

@hotdog123what321

Description

@hotdog123what321

The code currently only checks for the MPS backend which is for apple silicon only:
https://github.com/davepl/ObjectDetect/blob/main/detect.py#L20

You need to switch it to this to use your CUDA cores:

self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions