using CPU error - Githubissues

yya518 / FinBERT

A Pretrained BERT Model for Financial Communications. https://arxiv.org/abs/2006.08097

Apache License 2.0

560 stars 128 forks source link

using CPU error #23

Closed YeonGuKang closed 2 years ago

YeonGuKang commented 3 years ago

first, I really appreciate this model.

In my environment CUDA is available, but if i run the model, the process will die since using only CPU. Is there a separate setting method for using CUDA?

Thank you again for this model.