风格迁移模型性能测试

        为了优化模型的CPU占用以及运行时间,使用TensorFlow提供的测试脚本benchmark_model进行模型分析:

bazel-bin/tensorflow/tools/benchmark/benchmark_model \
--graph=/home/czczcz/test/mosaic512tv2Q.pb --input_layer="input" \
--input_layer_shape="1,299,299,3" --input_layer_type="float" \
--output_layer="output" --show_run_order=true --show_time=true \
--show_memory=true --show_summary=true --show_flops=true

        得到的结果如下:


性能测试.png
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容