Stack-Captioning
Stack-Captioning copied to clipboard
Possible to include a pre-trained model on CPU?
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!