Merge pull request #63 from secsilm/master

Fix comment errors in NNLM
graykode · May 2, 2021 · 8ec1aeb · 8ec1aeb
2 parents f97cef9 + af0b28a
commit 8ec1aeb
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/1-1.NNLM/NNLM.ipynb b/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 --git a/1-1.NNLM/NNLM.py b/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