he-transformer with the GPU

pjattke / docker-he-transformer

1 stars 1 forks source link

he-transformer with the GPU #2

Closed acoj1993 closed 3 years ago

acoj1993 commented 3 years ago

Hi @pjattke,

Dd you maybe try running the library with the GPU? If yes, what changes did you have to make?

Cheers, Aleksandar

pjattke commented 3 years ago

Hi @acoj1993

Unfortunately, we did not try running it with the GPU as we were training very small models only. I would suggest opening a new ticket in the official IntelAI/he-transformer repo by mentioning fboemer. He was super helpful in the issues we had and I'm sure he will help you figuring this out too.

Best regards, Patrick

acoj1993 commented 3 years ago

Aha, OK. Thanks for the suggestion, will do.

Cheers, Aleksandar