Skip to content

GitLab

Commit bf9c917a authored by Neal Wu's avatar Neal Wu Committed by GitHub
Browse files

Merge pull request #1870 from korrawat/export_inf_graph 

Change split name in slim's export_inference_graph
parents 2d5b39ad 1c704487
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
slim/export_inference_graph.py
View file @ bf9c917a
...@@ -100,7 +100,7 @@ def main(_): ...@@ -100,7 +100,7 @@ def main(_):
raise ValueError('You must supply the path to save to with --output_file') raise ValueError('You must supply the path to save to with --output_file')
tf.logging.set_verbosity(tf.logging.INFO) tf.logging.set_verbosity(tf.logging.INFO)
with tf.Graph().as_default() as graph: with tf.Graph().as_default() as graph:
dataset = dataset_factory.get_dataset(FLAGS.dataset_name, 'validation', dataset = dataset_factory.get_dataset(FLAGS.dataset_name, 'train',
FLAGS.dataset_dir) FLAGS.dataset_dir)
network_fn = nets_factory.get_network_fn( network_fn = nets_factory.get_network_fn(
FLAGS.model_name, FLAGS.model_name,
......
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