benchmarks icon indicating copy to clipboard operation
benchmarks copied to clipboard

Exporting tf cnn benchmarks to frozen pb graph

Open kahmed10 opened this issue 6 years ago • 1 comments

Is it possible to export the cnn benchmarks graphs using tensorflow's freeze_graph script? If so, what are the appropriate arguments to pass into tf_cnn_benchmarks.py and freeze_graph.py?

kahmed10 avatar Mar 27 '19 19:03 kahmed10

In principle, this is possible, but we have not tried this internally so we cannot provide much help. Any run of tf_cnn_benchmarks should work as long you pass --train_dir, --graph_file. and do not pass --eval. I think freeze_graph can then be run, by passing --input_graph with the graph file and --input_checkpoint with the checkpoint inside the train dir.

reedwm avatar Apr 02 '19 00:04 reedwm