Merge pull request #71 from majianjia/dev

Update main.py
majianjia · Nov 27, 2019 · a463162 · a463162
2 parents 5536110 + 40fbf68
commit a463162
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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()
+    main()