Open HAMZA310 opened 1 year ago
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).
torch.arange(...)
words[:5]
prob
Hey Andrej, This fixes a shape mismatch error thrown in
torch.arange(...)
. It seems like in the video it'swords[:5]
(hence 32 rows inprob
) but in the final notebook, it's all words (hence >32 rows inprob
).