Skip to content

Instantly share code, notes, and snippets.

@petewarden

petewarden/benchmark.sh

Created July 24, 2017 19:47
Show Gist options
  • Save petewarden/7ff91472455738c8f45b39e7f3777823 to your computer and use it in GitHub Desktop.
Save petewarden/7ff91472455738c8f45b39e7f3777823 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/inception_graph.pb --input_layer="Mul:0" \
--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