amzn / convolutional-handwriting-gan

ScrabbleGAN: Semi-Supervised Varying Length Handwritten Text Generation (CVPR20)
https://www.amazon.science/publications/scrabblegan-semi-supervised-varying-length-handwritten-text-generation
MIT License
264 stars 55 forks source link

if no gpu, use cpu when generating words #9

Closed myrriad closed 3 years ago

myrriad commented 3 years ago

Issue #, if available: Description of changes: If there is no gpu, use the cpu in generate_wordsLMDB.py

I copied and pasted this fix from a modified, working version in Google Colab, but I haven't actually tested this exact file.

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.