New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Update image_classification_using_transfer_learning_in_pytorch.ipynb #516

Open

kooscode wants to merge 1 commit into spmallick:master from kooscode:patch-1

kooscode commented Nov 14, 2020

Num classes should not be entries -1

Otherwise last layer does not match input labels and you will get this error

/pytorch/aten/src/THCUNN/ClassNLLCriterion.cu:108: cunn_ClassNLLCriterion_updateOutput_kernel: block: [0,0,0], thread: [7,0,0] Assertion `t >= 0 && t < n_classes` failed.```


          Update image_classification_using_transfer_learning_in_pytorch.ipynb

9596b6e

Num classes should not be entries -1

kooscode mentioned this pull request

BUG - image-classification-using-transfer-learning-in-pytorch #517

Open

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment