aboutsummaryrefslogtreecommitdiff
path: root/lenet.py
diff options
context:
space:
mode:
authorVasil Zlatanov <v@skozl.com>2019-02-27 18:54:05 +0000
committerVasil Zlatanov <v@skozl.com>2019-02-27 18:54:05 +0000
commitf4344da5fe4eec07efd3a1cee245975402afcfb8 (patch)
tree8ac6d2dd85fc5e4cb431a7aa07866345040e8838 /lenet.py
parent6b5e772d9b047d8ae44bd120f6b297596e2ce463 (diff)
downloade4-gan-f4344da5fe4eec07efd3a1cee245975402afcfb8.tar.gz
e4-gan-f4344da5fe4eec07efd3a1cee245975402afcfb8.tar.bz2
e4-gan-f4344da5fe4eec07efd3a1cee245975402afcfb8.zip
Use == instead of = err
Diffstat (limited to 'lenet.py')
-rw-r--r--lenet.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/lenet.py b/lenet.py
index 93a80cf..fbcdd5d 100644
--- a/lenet.py
+++ b/lenet.py
@@ -112,7 +112,7 @@ def train_classifier(x_train, y_train, x_val, y_val, batch_size=128, epochs=100,
model = get_lenet(shape)
- if optimizer = None:
+ if optimizer == None:
optimizer = optimizers.SGD(lr=0.001, decay=1e-6, momentum=0.9, nesterov=True)
model.compile(loss='categorical_crossentropy', metrics=metrics, optimizer=optimizer)