diff options
Diffstat (limited to 'lenet.py')
-rw-r--r-- | lenet.py | 2 |
1 files changed, 1 insertions, 1 deletions
@@ -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) |