我正在使用tf.Estimator在Tensorflow中训练和评估我的模型，我想知道在导出模型时批量大小是否等于1，换句话说，如果模型可以以预测模式导出？
Estimator默认创建checkpoint，.index和.meta文件，可以使用.pbtxt文件，我可以使用它获得protobuf模型，但问题是在该图表中批量大小是硬编码的，这不是我的东西想 . 我正在使用 tf.estimator.inputs.numpy_input_fn 将数据输入到我的模型中，并且给我的模型的输入张量具有维度[Batch_size，height，width，channels]，因为该函数负责实例化输入张量 .