kmkurn / pytorch-crf

(Linear-chain) Conditional random field in PyTorch.
https://pytorch-crf.readthedocs.io
MIT License
943 stars 152 forks source link

why don't support GPU? #67

Closed flyangovoyang closed 4 years ago

flyangovoyang commented 4 years ago

as title said. the error message says, cpu device expected but cuda is got

kmkurn commented 4 years ago

The library definitely supports GPU. Do you have a minimal sample code that reproduces the error?