diff options
| author | Vasil Zlatanov <v@skozl.com> | 2019-03-08 01:07:16 +0000 |
|---|---|---|
| committer | Vasil Zlatanov <v@skozl.com> | 2019-03-08 01:07:16 +0000 |
| commit | 9818b0b8283951526b9e06ba6ccb6de5ee080d37 (patch) | |
| tree | d5927682602b1dfe0053c3159dfd0442575db61e /lenet.py | |
| parent | 2008dbfa3bb70543b8d071947e3877b01b730308 (diff) | |
| download | e4-gan-9818b0b8283951526b9e06ba6ccb6de5ee080d37.tar.gz e4-gan-9818b0b8283951526b9e06ba6ccb6de5ee080d37.tar.bz2 e4-gan-9818b0b8283951526b9e06ba6ccb6de5ee080d37.zip | |
Use softmax back again in training lenet
Diffstat (limited to 'lenet.py')
| -rw-r--r-- | lenet.py | 3 |
1 files changed, 1 insertions, 2 deletions
@@ -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): |