Problem with makemore_part2_mlp.ipynb

[dewijones92]

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]

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

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