Hi! 👋
In the notebook 01 - Word2Vec (Skipgram).ipynb, the Skipgram class defines two separate embedding layers:
self.embedding_center = nn.Embedding(voc_size, emb_size)
self.embedding_outside = nn.Embedding(voc_size, emb_size)
However, in the forward() method, only embedding_center is used — even for the outside words and vocab logits:
outside_embedding = self.embedding_center(outside)
all_vocabs_embedding = self.embedding_center(all_vocabs)
It seems this might just be a small typo. Should these lines instead use self.embedding_outside? Like this:
outside_embedding = self.embedding_outside(outside)
all_vocabs_embedding = self.embedding_outside(all_vocabs)
Thanks so much for the great notebook! Really enjoying it. 🚀
Hi! 👋
In the notebook 01 - Word2Vec (Skipgram).ipynb, the Skipgram class defines two separate embedding layers:
self.embedding_center = nn.Embedding(voc_size, emb_size)
self.embedding_outside = nn.Embedding(voc_size, emb_size)
However, in the forward() method, only embedding_center is used — even for the outside words and vocab logits:
It seems this might just be a small typo. Should these lines instead use self.embedding_outside? Like this:
Thanks so much for the great notebook! Really enjoying it. 🚀