Skip to content

Instantly share code, notes, and snippets.

@petewarden

petewarden/benchmark.sh

Created July 24, 2017 19:48
Show Gist options
  • Save petewarden/462677056af0e9a8c2fd55a9cad2935b to your computer and use it in GitHub Desktop.
Save petewarden/462677056af0e9a8c2fd55a9cad2935b to your computer and use it in GitHub Desktop.
Download ZIP
Raw
benchmark.sh
bazel build -c opt tensorflow/tools/benchmark:benchmark_model \
&& bazel-bin/tensorflow/tools/benchmark/benchmark_model \
--graph=/tmp/tensorflow_inception_graph.pb \
--input_layer="Mul" --input_layer_shape="1,299,299,3" \
--input_layer_type="float" --output_layer="softmax:0" \
--show_run_order=false --show_time=false \
--show_memory=false --show_summary=true \
--show_flops=true --logtostderr
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment