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