karpathy/nn-zero-to-hero

Problem with makemore_part2_mlp.ipynb

dewijones92 opened this issue · 1 comments

dewijones92 commented 
loss = -prob[torch.arange(32), Y].log().mean()
loss
---------------------------------------------------------------------------
IndexError                                Traceback (most recent call last)
Cell In[20], line 1
----> 1 loss = -prob[torch.arange(32), Y].log().mean()
      2 loss

IndexError: shape mismatch: indexing tensors could not be broadcast together with shapes [32], [228146]
dewijones92 commented

btw, I have a fix
I will make a PR now

thanks so much for this AWESEOME video series @karpathy :)

  • 👍1