美文网首页
风格迁移模型性能测试

风格迁移模型性能测试

作者: nonoka | 来源:发表于2019-01-23 14:44 被阅读0次

            为了优化模型的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

    相关文章

      网友评论

          本文标题:风格迁移模型性能测试

          本文链接:https://www.haomeiwen.com/subject/idgjjqtx.html