@ptrblck Thank you for taking a look! I just recently started working with embeddings so I want to make sure I understand your comment.
In the model.train() section (below) the
for i,e in enumerate(self.all_embeddings):
    embeddings.append(e(x_categorical[:,i]))

takes in embeddings of these sizes

image1275×313 6.21 KB

In the model.eval() section, the embeddings are this size
torch.Size([3, 1])

Is there a different way to check the shape of self.all_embeddings?