From 9818b0b8283951526b9e06ba6ccb6de5ee080d37 Mon Sep 17 00:00:00 2001 From: Vasil Zlatanov Date: Fri, 8 Mar 2019 01:07:16 +0000 Subject: Use softmax back again in training lenet --- lenet.py | 3 +-- 1 file changed, 1 insertion(+), 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): -- cgit v1.2.3-54-g00ecf