benchmarks
benchmarks copied to clipboard
Exporting tf cnn benchmarks to frozen pb graph
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?
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.