"examples/hubert/vscode:/vscode.git/clone" did not exist on "ffeba11aff8bc9f1c3f7b214da74e44f97f456ef"
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
......@@ -100,7 +100,7 @@ def main(_):
raise ValueError('You must supply the path to save to with --output_file')
tf.logging.set_verbosity(tf.logging.INFO)
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)
network_fn = nets_factory.get_network_fn(
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