Skip to content

Fix first arg in torch.arange#5

Open
HAMZA310 wants to merge 1 commit into
karpathy:masterfrom
HAMZA310:master
Open

Fix first arg in torch.arange#5
HAMZA310 wants to merge 1 commit into
karpathy:masterfrom
HAMZA310:master

Conversation

@HAMZA310
Copy link
Copy Markdown

Hey Andrej,
This fixes a shape mismatch error thrown in torch.arange(...). It seems like in the video it's words[:5] (hence 32 rows in prob) but in the final notebook, it's all words (hence >32 rows in prob).

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

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants