I found something peculiar.
Running the lesson1-pets.ipynb for example, I can visualize the heatmap without issues.
When I add mixup “learn = learn.mixup”, for some reason commands like learn.get_preds return a rank 0 tensor instead of a rank 1 tensor. This causes all downstream interpretation function calls to fail.
Is anyone seeing this?