如何将批量张量输入单值张量?

时间:2018-01-21 21:11:10

标签: tensorflow

尝试微调MobileNet_v1_1.0_224

我面临的问题是输入:0有一个形状[1,224,224,3]。我想给它喂一批任意大小,例如[100,224,224,3]。我正在玩的代码是Tensorflow for poetsretrain.py是准确的。

以下是从输入打印的内容:0 tensor:

image_buffer_input = graph.get_tensor_by_name('input:0')
print (image_buffer_input)
> Tensor("input:0", shape=(1, 224, 224, 3), dtype=float32)

尝试输送100批次会引发错误:

ValueError: Cannot feed value of shape (100, 224, 224, 3) for Tensor 'input:0', which has shape '(1, 224, 224, 3)'

有没有办法绕过它?

P.S。

以下是我尝试训练的方法:

  sess.run(train_step,
        feed_dict={image_buffer_input: train_image_buffers,
                   ground_truth_input: train_ground_truth})

0 个答案:

没有答案