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

Question

Keras逐批将数据加载到GPU上(作者在此处注明).

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

(几周前有人在Keras名单上问了同样的问题,但到目前为止还没有回复.)

Answer 1

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