Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

added CPU support to se_extractor.py #262

Open
wants to merge 1 commit into
base: main
Choose a base branch
from

Commits on Jun 5, 2024

  1. added CPU support to se_extractor.py

    implemented an if statement to check if device is cuda or cpu, if it is a GPU device use GPU for extracting, else use CPU. formerly there was no CPU extraction support by default. by doing this way we can utilize CPU power for small scale extraction without wasting GPU power
    Afnanksalal committed Jun 5, 2024
    Configuration menu
    Copy the full SHA
    905d587 View commit details
    Browse the repository at this point in the history