Open zhangyikaii opened 4 years ago
Excuse me, https://github.com/graykode/nlp-tutorial/blob/master/1-1.NNLM/NNLM-Torch.py#L50 The comment here may be wrong. It should be
X = X.view(-1, n_step * m) # [batch_size, n_step * m]
Sorry for disturbing you.
hey bro, I find this mistake too, I think you are right
Excuse me, https://github.com/graykode/nlp-tutorial/blob/master/1-1.NNLM/NNLM-Torch.py#L50 The comment here may be wrong. It should be
X = X.view(-1, n_step * m) # [batch_size, n_step * m]
Sorry for disturbing you.