Im2Text icon indicating copy to clipboard operation
Im2Text copied to clipboard

CPU deployments?

Open mattndu opened this issue 7 years ago • 3 comments

This library looks excellent. It works great when I trained with the GPU. Eventually I'll need to add the model into an app that won't have GPU capability. Is it possible to do forward-pass on a CPU? If not, could you provide any pointers for tweaks to make it work?

Thanks for open sourcing such a great project!

mattndu avatar Dec 30 '16 11:12 mattndu

Thanks for your interest! Currently we are using cudnn in the CNN part, so we cannot run on pure CPU environment. However, I think you can use cudnn.convert (https://github.com/soumith/cudnn.torch#conversion-between-cudnn-and-nn) to convert the cudnn part to nn.

da03 avatar Dec 31 '16 00:12 da03

@da03 In OpenNMT we have a tool called release_model.lua for exactly this use case. https://github.com/OpenNMT/OpenNMT/blob/master/tools/release_model.lua . Can we check if that works for im2text?

srush avatar Jan 01 '17 22:01 srush

Sure I'll check.

da03 avatar Jan 01 '17 22:01 da03