Commit 1048ffd5 authored by Taylor Robie's avatar Taylor Robie
Browse files

change eval_batch_size flag from a string to an int

parent ec0d43ba
......@@ -321,7 +321,7 @@ def define_ncf_flags():
name="download_if_missing", default=True, help=flags_core.help_wrap(
"Download data to data_dir if it is not already present."))
flags.DEFINE_string(
flags.DEFINE_integer(
name="eval_batch_size", default=None, help=flags_core.help_wrap(
"The batch size used for evaluation. This should generally be larger"
"than the training batch size as the lack of back propagation during"
......
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