aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorVasil Zlatanov <v@skozl.com>2019-03-08 01:07:16 +0000
committerVasil Zlatanov <v@skozl.com>2019-03-08 01:07:16 +0000
commit9818b0b8283951526b9e06ba6ccb6de5ee080d37 (patch)
treed5927682602b1dfe0053c3159dfd0442575db61e
parent2008dbfa3bb70543b8d071947e3877b01b730308 (diff)
downloade4-gan-9818b0b8283951526b9e06ba6ccb6de5ee080d37.tar.gz
e4-gan-9818b0b8283951526b9e06ba6ccb6de5ee080d37.tar.bz2
e4-gan-9818b0b8283951526b9e06ba6ccb6de5ee080d37.zip
Use softmax back again in training lenet
-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):