Dog Breed Identification (ImageNet Dogs) on Kaggle

[anirudh]

https://d2l.ai/chapter_computer-vision/kaggle-dog.html

[StevenJokess]

https://github.com/StevenJokess/d2l-en-read/blob/moreme/Ch13_CV/kaggle_dog%20(1).ipynb
Would you like to help me figure out why no test DIR when using colab:

[Chen_pelagius]

“output = torch.nn.functional.softmax(net(data.to(devices[0])), dim=0)

" dim" should be “1” but not “0” to ensure the sum of probability of all the classes to be 1?