aboutsummaryrefslogtreecommitdiff
path: root/lenet.py
diff options
context:
space:
mode:
authornunzip <np.scarh@gmail.com>2019-03-08 01:34:26 +0000
committernunzip <np.scarh@gmail.com>2019-03-08 01:34:26 +0000
commit59e5363b41a08b857d2246d51d144cf197a6d6e5 (patch)
treef04654d25acf0448bae6bf2802d5b36edfe2a503 /lenet.py
parentd36c8c88fc435aa4372e6c5564f0ff12fc7120e9 (diff)
parent9818b0b8283951526b9e06ba6ccb6de5ee080d37 (diff)
downloade4-gan-59e5363b41a08b857d2246d51d144cf197a6d6e5.tar.gz
e4-gan-59e5363b41a08b857d2246d51d144cf197a6d6e5.tar.bz2
e4-gan-59e5363b41a08b857d2246d51d144cf197a6d6e5.zip
Merge branch 'master' of skozl.com:e4-gan
Diffstat (limited to 'lenet.py')
-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):