如何使用* .pb模型在Tensorflow中重新训练变量

时间:2020-09-24 04:29:43

标签: python tensorflow

我有一个张紧的* .pb张量流模型,我想重新训练该模型,但只更新图中的一个变量(名为“ add_input”),最后我想要获得更新后的张量的值,如何明白了。

现在,我定义一个变量来馈送要更新的张量,像这样:

_add_input = sess.graph.get_tensor_by_name('add_input:0')
add_factor = tf.Variable(tf.zeros_like(_add_input))
sess.run(train_op, feed_dict={_add_input: add_factor})

不幸的是,我得到了错误:

ValueError: setting an array element with a sequence.

0 个答案:

没有答案
相关问题