Running Inference using CPU only.

ruotianluo / self-critical.pytorch

Unofficial pytorch implementation for Self-critical Sequence Training for Image Captioning. and others.

MIT License

991 stars 278 forks source link

Running Inference using CPU only. #274

Open harindercnvrg opened 2 years ago

harindercnvrg commented 2 years ago

I am not able to run inference on the model provided in the demo using CPU only. I am getting cuda device not available error. I tried removing all gpu and cuda references in the code and replacing it with CPU but it still does not work. It would be really helpful if you could help understand how to run the demo code without GPU.

Thanks.