tensorflow模型量化

最新推荐文章于 2025-04-17 17:45:33 发布

weixin_33766168

最新推荐文章于 2025-04-17 17:45:33 发布

阅读量441

点赞数

文章标签：人工智能

本文详细介绍了使用TensorFlow进行模型量化的步骤与方法，通过具体的命令行参数展示如何将预训练的ResNet100模型从浮点表示转换为更节省资源的量化表示，包括属性默认添加、节点移除、常数折叠、批量标准化融合等关键操作。

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

tensorflow模型量化
/DATA/share/DeepLearning/code/tensorflow/bazel-bin/tensorflow/tools/graph_transforms/transform_graph \
--in_graph=./model_resnet100.pb \
--out_graph=/tmp/model_resnet100_quantized_graph.pb \
--inputs=input0 \
--outputs=fcblock/fc1/add_1 \
--transforms='add_default_attributes strip_unused_nodes(type=float, shape="1,112,112,3")
remove_nodes(op=Identity, op=CheckNumerics) fold_constants(ignore_errors=true)
fold_batch_norms fold_old_batch_norms quantize_weights quantize_nodes
strip_unused_nodes sort_by_execution_order'