我正在使用带有数字的张量流。我想扩展服务功能的位数。
我已经可以通过训练在运行时为模型提供服务了,但要点是,我想在模型已经训练之后实现这种可能性。为此,我需要会话,输入和输出变量。
我正在从检查点文件中获取会话和输入。只是我有问题要获取输出。我知道输出会以常量形式存储在冻结图中。
freeze_graph.freeze_graph(
input_graph=graphdef_path,
input_saver='',
input_binary=True,
input_checkpoint=checkpoint_path,
**output_node_names=out_name**,
restore_op_name="save/restore_all",
filename_tensor_name="save/Const:0",
output_graph=path_frozen,
clear_devices=True,
initializer_nodes="",
)
但是我不知道如何获取它们以及是否可以通过常量build_tensor_info。
谢谢,谢谢。