Dog Breed Identification (ImageNet Dogs) on Kaggle

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

1 Like

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:


“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?

Indeed that’s been fixed just now. Thanks for pointing it out :slight_smile: