Skip to content
This repository has been archived by the owner on May 11, 2024. It is now read-only.

he-transformer with the GPU #62

Open
acoj1993 opened this issue Feb 19, 2021 · 3 comments
Open

he-transformer with the GPU #62

acoj1993 opened this issue Feb 19, 2021 · 3 comments

Comments

@acoj1993
Copy link

Hey @fboemer,

Is it possible to run the library with the GPU? If yes, what changes do I have to make?

Cheers,
Aleksandar
@fboemer
Copy link
Contributor

fboemer commented Feb 19, 2021

Hi Aleksandar, thanks for your interest. Unfortunately this library supports CPU only.

@acoj1993
Copy link
Author

Aha, ok, understood.

@acoj1993 acoj1993 reopened this Feb 20, 2021
@acoj1993
Copy link
Author

While I am at it, another question.

Have you tried other ImageNet models, apart from MobileNet V2? Do you have any instructions or something I should keep in mind?

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@fboemer @acoj1993