openai / pixel-cnn

Code for the paper "PixelCNN++: A PixelCNN Implementation with Discretized Logistic Mixture Likelihood and Other Modifications"
https://arxiv.org/abs/1701.05517
Other
1.92k stars 436 forks source link

Speeding up image generation up to 183x #10

Open PrajitR opened 7 years ago

PrajitR commented 7 years ago

As currently implemented, generation speed significantly drops for bigger batch sizes (i.e. ~11 minutes for 16 images on a K40). Some of my friends and I recently worked on improving generation speed through smart caching. We are able to achieve up to a 183x speedup for generation. The repo and our ICLR workshop submission has more details.

Is there any interest in merging some of these changes into the main repository?