tensorflow 模型量化

本文介绍了一种无需使用TFLite的TensorFlow模型量化方法。通过官方提供的量化脚本quantize_graph.py,可以将32位浮点型模型转换为8位整型模型,显著减小模型大小并提高运行效率。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

文章参考https://blog.youkuaiyun.com/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
 

评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值