Keras将数据批量加载到GPU上(作者注here https://github.com/fchollet/keras/issues/249).
对于小型数据集,这是非常低效的。有没有办法修改 Keras 或直接调用 Theano 函数(在 Keras 中定义模型之后)以允许将所有批次预先移动到 GPU,并使用 GPU 内存中已有的批次完成训练?
(有人问同样的问题在 Keras 列表中 https://groups.google.com/forum/#!topic/keras-users/1yANBtgJMuE几周前,但到目前为止还没有回复。)
只需将数据作为不可训练的嵌入矩阵硬连接到模型中(带有自定义初始值设定项的嵌入层)。然后,您将一堆索引传递给 model.fit 方法,而不是训练数据。
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)