aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--lenet.py3
1 files changed, 1 insertions, 2 deletions
diff --git a/lenet.py b/lenet.py
index 4950fe9..6bea8dd 100644
--- a/lenet.py
+++ b/lenet.py
@@ -64,8 +64,7 @@ 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 = 'relu'))
+ model.add(Dense(units=10, activation = 'softmax'))
return model
def get_lenet_icp(shape):