Gpu predicts switching to Cpu

kanyun-inc / fairseq-gec

Source code for paper: Improving Grammatical Error Correction via Pre-Training a Copy-Augmented Architecture with Unlabeled Data

Other

245 stars 67 forks source link

Closed 1228589545 closed 4 years ago

1228589545 commented 4 years ago

Hello, I want to change the GPU to CPU and change those parameters during the prediction and training？

zhawe01 commented 4 years ago

I haven't tried that, but little code change should work.