From 0d2a5c099fdfd3ef122c0ef371b622478a3a8c08 Mon Sep 17 00:00:00 2001
From: Seohee Moon <moonsh0913@gmail.com>
Date: Fri, 5 Jan 2024 04:36:41 +0000
Subject: [PATCH] added missing non-linearity

---
 mnist/main.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/mnist/main.py b/mnist/main.py
index 1bee55c4..8f0bc49f 100644
--- a/mnist/main.py
+++ b/mnist/main.py
@@ -20,8 +20,9 @@ def __init__(self):
 
     def forward(self, x):
         x = self.conv1(x)
-        x = F.relu(x)
+		x = F.relu(x)
         x = self.conv2(x)
+		x = F.relu(x)
         x = F.max_pool2d(x, 2)
         x = self.dropout1(x)
         x = torch.flatten(x, 1)