diff --git a/examples/auto_test/main.py b/examples/auto_test/main.py index c07698b..0ca3ebe 100644 --- a/examples/auto_test/main.py +++ b/examples/auto_test/main.py @@ -66,7 +66,7 @@ def build_model(x_shape): def train(model, x_train, y_train, x_test, y_test, batch_size=64, epochs=50): # save best checkpoint = ModelCheckpoint(filepath=save_dir, - monitor='val_acc', + monitor='val_accuracy', verbose=0, save_best_only='True', mode='auto', @@ -157,4 +157,4 @@ def main(): raise Exception('test failed, accuracy is %.1f%% < 80%%' % (acc * 100.0)) if __name__ == "__main__": - main() \ No newline at end of file + main()