我训练了小型模型,并使用tf.train.write_graph
保存了图形。现在,我有检查点文件和.pbtxt
文件。
我尝试使用
freeze_graph.freeze_graph(input_graph='freeze/Model.pbtxt',
input_saver="",
input_binary=False,
input_checkpoint='freeze/Model.ckpt',
output_node_names="conv2d/BiasAdd",
restore_op_name="save/restore_all",
filename_tensor_name='',
output_graph='frozenModel.pb',
clear_devices=True,
initializer_nodes=''
)
但是它将IndexError: list index out of range
扔到了initializer_nodes =''行中。
如果我删除此行,则会抛出TypeError: freeze_graph() missing 1 required positional argument: 'initializer_nodes'.
我找不到关于freeze_graph
参数的任何有意义的解释。如何解决这个错误?在freeze_graph
中,尤其是在initializer_nodes
中传递什么值?