modify code for working in gpu context. (#13302)

apache · Nov 17, 2018 · 64657c2 · 64657c2
1 parent 98830d5
commit 64657c2
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/docs/tutorials/python/predict_image.md b/docs/tutorials/python/predict_image.md
@@ -69,6 +69,7 @@ def get_image(url, show=False):
     img = mx.image.imresize(img, 224, 224) # resize
     img = img.transpose((2, 0, 1)) # Channel first
     img = img.expand_dims(axis=0) # batchify
+    img = img.astype('float32') # for gpu context
     return img
 
 def predict(url):