nqg
nqg copied to clipboard
Can we run it on CPU machine (No GPU)
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.