Stack-Captioning icon indicating copy to clipboard operation
Stack-Captioning copied to clipboard

Possible to include a pre-trained model on CPU?

Open yfeng997 opened this issue 6 years ago • 0 comments

I am wondering if it's possible to include a checkpoint for the model that is able to run on a CPU? I find the results coming out of this paper really amazing, and I want to generate some captions for my images. Many thanks!

yfeng997 avatar Dec 23 '18 15:12 yfeng997