theano - 如何将所有批量数据加载到 Keras(Theano 后端)的 GPU 内存中？

Question

Keras 将数据批量加载到 GPU 上(作者注明here)。

对于小型数据集，这是非常低效的。有没有办法修改 Keras 或直接调用 Theano 函数(在 Keras 中定义模型之后)以允许将所有批处理预先移动到 GPU，并使用 GPU 内存中已有的批处理完成训练？

(几周前有人问过同样的问题on the Keras list，但到目前为止还没有回复。)

Answer 1

只需将数据作为不可训练的嵌入矩阵硬连接到模型中(带有自定义初始值设定项的嵌入层)。然后，您将一堆索引传递给 model.fit 方法，而不是训练数据。