为什么我的输入张量的batch_size受约束(tensorflow)?

时间:2019-03-12 06:06:15

标签: python tensorflow

我正在尝试训练mnist分类器,并下载了源代码here。我将第31行中的op命名为“ input_tensor”。

然后我使用freeze_graphoptimize_for_inference将其转换为opt_mnist_graph.pb

但是在tensorboard中,opt_mnist_graph.pb显示input_tensor的batch_size被限制为100,例如 this

在cnn_model_fn()函数中没有batch_size,在训练模型时我仅将batch_size = 100设置为。

那么为什么我的input_tensor的batch_size为100,以及如何使batch_size不受限制,以便可以预测单个图像?

0 个答案:

没有答案