批量大小如何在 TimeDistributed 中工作

Question

我是 AI 初学者，正在尝试在 Keras 中实现 CRNN 模型。

model.add(TimeDistributed(base_model, input_shape=(3,32,32,3)))

我知道上面的代码创建了 3 个时间步并使用 32x32 RGB 图像。

那么，如果我有 90 个 train_image 并将批量大小设置为 30，它是如何工作的？

grouped 30 pieces and entered into the timestep

或

enter into timestep in order

还是我对批量大小有误解？

Answer 1

如果您有 90 张图像并且希望批量大小为 30，那么您的 input_shape 应该是 (90,30,32,32,3)。来源：文档https://keras.io/api/layers/recurrent_layers/time_distributed/

批量大小是在 CRNN 更新其内部参数之前，您希望在学习迭代中使用的样本数量。

正如您在屏幕截图中看到的那样，您已经在 3 个时间步中训练了一个时期的模型（在训练您的学习模型时，一个时期是对整个数据集的一次迭代。30 乘以 3 等于 90，整个数据集） .