Fix comment errors in NNLM.py

Author: secsilm

Diff for: 1-1.NNLM/NNLM.ipynb

@@ -35,8 +35,8 @@
         "        self.b = nn.Parameter(torch.ones(n_class))\n",
         "\n",
         "    def forward(self, X):\n",
-        "        X = self.C(X) # X : [batch_size, n_step, n_class]\n",
-        "        X = X.view(-1, n_step * m) # [batch_size, n_step * n_class]\n",
+        "        X = self.C(X) # X : [batch_size, n_step, m]\n",
+        "        X = X.view(-1, n_step * m) # [batch_size, n_step * m]\n",
         "        tanh = torch.tanh(self.d + self.H(X)) # [batch_size, n_hidden]\n",
         "        output = self.b + self.W(X) + self.U(tanh) # [batch_size, n_class]\n",
         "        return output\n",

Diff for: 1-1.NNLM/NNLM.py

@@ -30,8 +30,8 @@ def __init__(self):
         self.b = nn.Parameter(torch.ones(n_class))
 
     def forward(self, X):
-        X = self.C(X) # X : [batch_size, n_step, n_class]
-        X = X.view(-1, n_step * m) # [batch_size, n_step * n_class]
+        X = self.C(X) # X : [batch_size, n_step, m]
+        X = X.view(-1, n_step * m) # [batch_size, n_step * m]
         tanh = torch.tanh(self.d + self.H(X)) # [batch_size, n_hidden]
         output = self.b + self.W(X) + self.U(tanh) # [batch_size, n_class]
         return output