Dog Breed Identification (ImageNet Dogs) on Kaggle

1 Like
Would you like to help me figure out why no test DIR when using colab:

“output = torch.nn.functional.softmax(net([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: