Thilina Rajapakse
1 min readOct 30, 2019

--

model = TransformerModel('roberta', 'roberta-base', use_cuda=False, args={'fp16': False})

You have to set both use_cuda and fp16 to False as shown above. I should probably set fp16 to False if cuda is not used by default, but this should work for now.

--

--

Thilina Rajapakse

AI researcher, avid reader, fantasy and Sci-Fi geek, and fan of the Oxford comma. www.linkedin.com/in/t-rajapakse/