he-transformer with the GPU

[acoj1993]

Hey @fboemer,

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

Cheers,
Aleksandar

[fboemer]

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

[acoj1993]

Aha, ok, understood.

[acoj1993]

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?