fixed small errors

5 years ago · ca2b2edae6
parent 9bc4b27945
commit ca2b2edae6
1 changed files with 1 additions and 1 deletions
--- a/classifier/nn_ff.py
+++ b/classifier/nn_ff.py
@ -79,7 +79,7 @@ class FeedForward(torch.nn.Module):
                self.optimizer.zero_grad()

                # forward pass
-                y_pred = self.forward(X_train[a:b])
+                y_pred = self.forward(self.X_train[a:b])
                loss = self.criterion(y_pred, self.y_train[a:b])

                # backward pass