tensorflow 模型量化

文章参考https://blog.csdn.net/u012101561/article/details/82587806?tdsourcetag=s_pcqq_aiomsg


tensorflow的量化有很多种方式,有些需要用到tflite,有些不需要,下面介绍一下不需要tflite也能量化的方式:

在较早的版本上有官方的一个量化脚本,比如tensorflow-1.10.0\tensorflow\tools\quantization\quantize_graph.py


假设你想把32bit浮点型的a.pb量化成8bit整型,命名为b.pb,只需在命令行中输入:

python  quantize_graph.py  --input=a.pb  --output_node_names=out1,out2,out3  --output=b.pb  --mode=eightbit

全部评论

相关推荐

点赞 收藏 评论
分享
牛客网
牛客企业服务