在训练阶段我在tensorflow中使用batch =5，为什么我不能在tensorflowjs中只使用batch =1测试？

Question

我使用batchsize=5在tensorflow中训练GAN模型，因此生成器的输入大小为5，图像大小，图像大小，3。训练后，我将tensorflow模型转换为tensorflowjs模型。

因此，我使用loadFrozenModel.Then model.predict来加载模型来预测image.However，model.execute(dict)中提供的dict‘’concat‘的形状必须是5,512,512,12，而不是1,512,512,12。如何解决这个问题?我在tensorflow中的训练阶段使用mini-batch，在tensorflow中只预测一个输入的图像，而不是tensorflow中的5个输入

Figure 1. the error

Answer 1

听起来像是您在训练作业中明确地将批处理大小设置为输入形状的一部分，例如

x = tf.placeholder("float", shape=[5, 512, 512, 12])

相反，您应该不指定批处理大小，如下所示：

x = tf.placeholder("float", shape=[None, 512, 512, 12])

这样，无论在训练时还是在推理时，图形都将与您提供的任何批处理大小一起工作。

如果您的代码需要显式地知道批处理大小，请参阅here获取一些技巧。