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) | 
