Merge branch 'cifar_keras' of https://github.com/tensorflow/models into cifar_keras

87c0e09d · Priya Gupta · 3060b534 · e17e64bb · 87c0e09d
Commit 87c0e09d authored Dec 07, 2018 by Priya Gupta
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

official/resnet/keras/keras_cifar_main.py official/resnet/keras/keras_cifar_main.py +2 -0

No files found.
--- a/official/resnet/keras/keras_cifar_main.py
+++ b/official/resnet/keras/keras_cifar_main.py
@@ -301,6 +301,8 @@ def run_cifar_with_keras(flags_obj):
              lr_callback,
              tesorboard_callback
            ],
+            validation_steps=num_eval_steps,
+            validation_data=eval_input_dataset,
            verbose=1)
  
  eval_output = model.evaluate(eval_input_dataset,