model.fit 和 model.predict 中的批量大小

Question

在 中keras，model.fit和model.predict都有一个参数batch_size。我的理解是 model.fit 中的批量大小与批量优化有关，batch_sizein的物理含义是model_predict什么？它是否需要等于 使用的那个model.fit？

Answer 1

不，它没有。想象一下，在您的模型中，有一个函数可以显着增加内存量。因此，如果您尝试一次性预测所有数据，则可能会遇到资源错误。当您使用 GPU 内存有限的 GPU 进行预测时，通常会出现这种情况。因此，您选择同时只预测小批量。predict 函数中的 batch_size 参数不会以任何方式改变您的结果。所以你可以选择任何你想要的 batch_size 进行预测。