From 3afe0286598841c709db79ac2112e0e6dd2c122b Mon Sep 17 00:00:00 2001 From: yelircaasi Date: Sat, 1 Aug 2020 13:48:39 +0200 Subject: [PATCH 1/3] 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 f478f1d..55ba991 100644 --- a/classifier/nn_ff.py +++ b/classifier/nn_ff.py @@ -58,7 +58,7 @@ class FeedForward(torch.nn.Module): self.samples1 = samples1 self.samples2 = samples2 - model.eval() # put into eval mode + self.eval() # put into eval mode # initialize training data self.shuffle() From 9bc4b279458708f6db8199fb0863cf8a29853d9a Mon Sep 17 00:00:00 2001 From: yelircaasi Date: Sat, 1 Aug 2020 13:50:43 +0200 Subject: [PATCH 2/3] 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 From ca2b2edae60ea4966c7ecaf69f2686083bcf394d Mon Sep 17 00:00:00 2001 From: yelircaasi Date: Sat, 1 Aug 2020 13:51:46 +0200 Subject: [PATCH 3/3] 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 0e761d8..3ccda08 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 = 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