From 9bc4b279458708f6db8199fb0863cf8a29853d9a Mon Sep 17 00:00:00 2001
From: yelircaasi <isaac.r.riley@gmail.com>
Date: Sat, 1 Aug 2020 13:50:43 +0200
Subject: [PATCH] fixed small errors

---
 classifier/nn_ff.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/classifier/nn_ff.py b/classifier/nn_ff.py
index 55ba991..0e761d8 100644
--- 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 = model(X_train[a:b])
+                y_pred = self.forward(X_train[a:b])
                 loss = self.criterion(y_pred, self.y_train[a:b])
 
                 # backward pass