发新帖

TF-TRT的INT8量化问题

[复制链接]
129 0

快来加入 TensorFlowers 大家庭!

您需要 登录 才可以下载或查看,没有帐号?加入社区

x
大家好:
根据https://docs.nvidia.com/deeplearning/frameworks/tf-trt-user-guide/index.html#post-train的教程对Tensorflow的模型用TF-TRT进行了INT8量化,在进行量化校准时出现错误:
代码:
dataset = tf.data.TFRecordDataset(tf_calib_data_files)
iterator = dataset.make_one_shot_iterator()
features = iterator.get_next()
def input_map_fn():
    return {'input:0': features}

...

converted_graph_def = converter.calibrate(
    fetch_names=['detection_boxes_l1:0','detection_scores_l1:0','detection_classes_l1:0','detection_boxes_l2:0','detection_scores_l2:0','detection_classes_l2:0'],
    num_runs=10,
    input_map_fn=input_map_fn)
错误:
ValueError: node 'IteratorGetNext' in input_map does not exist in graph (input_map entry: input:0->IteratorGetNext:0)



我知道答案 回答被采纳将会获得10 金币 + 5 金币 已有0人回答
本楼点评(0) 收起
您需要登录后才可以回帖 登录 | 加入社区

本版积分规则

快速回复 返回顶部 返回列表