fastai/course-v3

In lesson2-download, category is not printed

andresesfm opened this issue · 1 comments

Instead what gets printed is:
Category tensor(0)
Expected: it should print the category such as:
Teddy
This has been addressed here:
#530

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)