fixed small errors

5 years ago · 3afe028659
parent 6e0fd99357
commit 3afe028659
1 changed files with 1 additions and 1 deletions
--- 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()