transformer-lm icon indicating copy to clipboard operation
transformer-lm copied to clipboard

Published 20 hours ago • verified publisher icon lopuhin

Predict with GPU

Open binhvq opened this issue 5 years ago • 0 comments

Hi lopuhin! I'm use GPU for trainning but i can't use GPU for text generate. i'm try change device in torch but recived message RuntimeError: Expected object of backend CUDA but got backend CPU for argument 3 'index'

binhvq avatar May 14 '19 06:05 binhvq