From 3afe0286598841c709db79ac2112e0e6dd2c122b Mon Sep 17 00:00:00 2001 From: yelircaasi Date: Sat, 1 Aug 2020 13:48:39 +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 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()