训练和测试阶段batchsize不一致怎么办?

ran
 
我用tensorflow,训练阶段的batch size是64,现在模型已经训练好了。但是我现在预测的时候,我先加载训练好的模型,然后对单个样本进行预测,但是会报错,输入数据shape不一致。我试过,只有把batch size也设置成64才能运行。
请问有什么办法能够让预测阶段的batch size与测试阶段不同?
谢谢!
已邀请:

tensorflow - CaffeCN社区Tensorflow主题站管理员 http://tensorflow.caffecn.cn

赞同来自:

shuju
一般来说,涉及到batch size的输入数据,在声明placeholder的时候会用None来代替,然后在run的时候传入batch size。

要回复问题请先登录注册