xinyadu / nqg

neural question generation for reading comprehension

Home Page:https://arxiv.org/abs/1705.00106

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Can we run it on CPU machine (No GPU)

karam86 opened this issue · comments

I don't have GPU machine. When I try to run the testing using the existing trained model. It given me error "cutorch" not installed. And for cutorch I need CUDA and for CUDA I need gpu machine. Is there a way around??

You can solve this problem by manually modifying the config-trans file in the model dir.
Simply:

src=../data/processed/src-dev.txt
beam_size=3
fallback_to_cpu=true
replace_unk=true

then you can run prediction on your CPU machine.
Hope this would help you.

Remove "gpuid=1" in the config-trans (for testing) and config-train (for training) files. After this change it works.
@karam86 @Aria-K-Alethia

Indeed, or change the gpuid to 0 and it will use CPU.