1 files changed, 2 insertions, 1 deletions
diff --git a/lenet.py b/lenet.py
index 5ed6705..97479ed 100644
--- a/lenet.py
+++ b/lenet.py
@@ -64,7 +64,8 @@ def get_lenet(shape):
   
   model.add(Dense(units=120, activation='relu'))
   model.add(Dense(units=84, activation='relu'))
-  model.add(Dense(units=10, activation = 'softmax'))
+  #model.add(Dense(units=10, activation = 'softmax'))
+  model.add(Dense(units=10, activation = 'relu'))
   return model
 
 def plot_history(history, metric = None):