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

作者: 谦成
发布时间: 2023-05-03 04:14:58 (1年前)
转自：

2 条回复

0#
回复此人
哎？小查查 | 2019-08-31 10-32

<div class =“post-text”itemprop =“text”> <P> 听起来您将批量大小明确地设置为训练作业中输入形状的一部分，例如 </p> <pre> <code> x = tf.placeholder("float", shape=[5, 512, 512, 12]) </code> </pre> <P> 相反，您应该不指定批量大小，如下所示： </p> <pre> <code> x = tf.placeholder("float", shape=[None, 512, 512, 12]) </code> </pre> <P> 这样，无论是在培训还是在推理时，图表都可以使用您提供的任何批量大小。 </p> <P> 如果您的代码需要明确知道批量大小，请参阅 <a href="https://www.tensorflow.org/guide/faq#how_do_i_build_a_graph_that_works_with_variable_batch_sizes" rel="nofollow noreferrer"> 这里 </A> 一些提示。 </p> </DIV>

编辑

登录后才能参与评论