In lesson2-download, category is not printed
andresesfm opened this issue · 1 comments
andresesfm commented
Instead what gets printed is:
Category tensor(0)
Expected: it should print the category such as:
Teddy
This has been addressed here:
#530
cbadjatya commented
The predict function returns 3 objects- prediction class, prediction id and output tensor.
You can get the class name by printing the prediction class.
pred_class,pred_idx,outputs = learn.predict(img); print(pred_class)