search

awslabs / handwritten-text-recognition-for-apache-mxnet

This repository lets you train neural networks models for performing end-to-end full-page handwriting recognition using the Apache MXNet deep learning frameworks on the IAM Dataset.
Apache License 2.0
481 stars 189 forks source link

Context set doesn't check for presence of GPU in 0_handwriting_ocr.ipynb #31

Closed fugumaki closed 4 years ago

fugumaki commented 4 years ago

When setting the context in the denoising section of 0_handwriting_ocr.ipynb, the code doesn't check for the presence of GPUs and therefore fails when no GPUs are available.

Based on other context setting code in the example, I would suggest that this line... ctx_nlp = mx.gpu(3)

Should read... ctx_nlp = mx.gpu(3) if mx.context.num_gpus() > 0 else mx.cpu()

jonomon commented 4 years ago

Hi @fugumaki Thank you for your comment. Please submit a PR for this.