Skip to content

GitLab

Commit d0c1b9da authored by Menglong Zhu's avatar Menglong Zhu Committed by dreamdragon
Browse files

Make eval input reader repeated field and update config util to handle this field. 

PiperOrigin-RevId: 211907984
parent 7a6b3acf
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
research/lstm_object_detection/utils/config_util_test.py
View file @ d0c1b9da
...@@ -45,7 +45,7 @@ class ConfigUtilTest(tf.test.TestCase): ...@@ -45,7 +45,7 @@ class ConfigUtilTest(tf.test.TestCase):
pipeline_config.train_config.batch_size = 32 pipeline_config.train_config.batch_size = 32
pipeline_config.train_input_reader.label_map_path = "path/to/label_map" pipeline_config.train_input_reader.label_map_path = "path/to/label_map"
pipeline_config.eval_config.num_examples = 20 pipeline_config.eval_config.num_examples = 20
pipeline_config.eval_input_reader.queue_capacity = 100 pipeline_config.eval_input_reader.add().queue_capacity = 100
pipeline_config.Extensions[ pipeline_config.Extensions[
internal_pipeline_pb2.lstm_model].train_unroll_length = 5 internal_pipeline_pb2.lstm_model].train_unroll_length = 5
...@@ -62,7 +62,7 @@ class ConfigUtilTest(tf.test.TestCase): ...@@ -62,7 +62,7 @@ class ConfigUtilTest(tf.test.TestCase):
configs["train_input_config"]) configs["train_input_config"])
self.assertProtoEquals(pipeline_config.eval_config, configs["eval_config"]) self.assertProtoEquals(pipeline_config.eval_config, configs["eval_config"])
self.assertProtoEquals(pipeline_config.eval_input_reader, self.assertProtoEquals(pipeline_config.eval_input_reader,
configs["eval_input_config"]) configs["eval_input_configs"])
self.assertProtoEquals( self.assertProtoEquals(
pipeline_config.Extensions[internal_pipeline_pb2.lstm_model], pipeline_config.Extensions[internal_pipeline_pb2.lstm_model],
configs["lstm_model"]) configs["lstm_model"])
...@@ -76,7 +76,7 @@ class ConfigUtilTest(tf.test.TestCase): ...@@ -76,7 +76,7 @@ class ConfigUtilTest(tf.test.TestCase):
pipeline_config.train_config.batch_size = 32 pipeline_config.train_config.batch_size = 32
pipeline_config.train_input_reader.label_map_path = "path/to/label_map" pipeline_config.train_input_reader.label_map_path = "path/to/label_map"
pipeline_config.eval_config.num_examples = 20 pipeline_config.eval_config.num_examples = 20
pipeline_config.eval_input_reader.queue_capacity = 100 pipeline_config.eval_input_reader.add().queue_capacity = 100
pipeline_config.Extensions[ pipeline_config.Extensions[
internal_pipeline_pb2.lstm_model].train_unroll_length = 5 internal_pipeline_pb2.lstm_model].train_unroll_length = 5
......
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