Unverified Commit 7e056690 authored by Toby Boyd's avatar Toby Boyd Committed by GitHub
Browse files

add validation_freq which matches estimator settings. (#6180)

- Modest speedup for CIFAR-10
- Slightly greater speedup expected for ImageNet ResNet50.
parent b6c0c7f9
...@@ -180,12 +180,13 @@ def run(flags_obj): ...@@ -180,12 +180,13 @@ def run(flags_obj):
], ],
validation_steps=num_eval_steps, validation_steps=num_eval_steps,
validation_data=validation_data, validation_data=validation_data,
validation_freq=flags_obj.epochs_between_evals,
verbose=2) verbose=2)
eval_output = None eval_output = None
if not flags_obj.skip_eval: if not flags_obj.skip_eval:
eval_output = model.evaluate(eval_input_dataset, eval_output = model.evaluate(eval_input_dataset,
steps=num_eval_steps, steps=num_eval_steps,
verbose=1) verbose=2)
stats = keras_common.build_stats(history, eval_output, time_callback) stats = keras_common.build_stats(history, eval_output, time_callback)
return stats return stats
......
...@@ -172,13 +172,14 @@ def run(flags_obj): ...@@ -172,13 +172,14 @@ def run(flags_obj):
], ],
validation_steps=num_eval_steps, validation_steps=num_eval_steps,
validation_data=validation_data, validation_data=validation_data,
validation_freq=flags_obj.epochs_between_evals,
verbose=2) verbose=2)
eval_output = None eval_output = None
if not flags_obj.skip_eval: if not flags_obj.skip_eval:
eval_output = model.evaluate(eval_input_dataset, eval_output = model.evaluate(eval_input_dataset,
steps=num_eval_steps, steps=num_eval_steps,
verbose=1) verbose=2)
stats = keras_common.build_stats(history, eval_output, time_callback) stats = keras_common.build_stats(history, eval_output, time_callback)
return stats return stats
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment