【发布时间】:2017-08-21 10:17:37
【问题描述】:
我的神经网络模型是使用 GPU 在 Theano 上的 Keras 中构建的。 我正在使用 Pickle 存储它以备将来使用,可能在另一台计算机上。
是否可以在没有 GPU 的情况下使用模型进行预测?
【问题讨论】:
标签: neural-network gpu theano pickle keras
我的神经网络模型是使用 GPU 在 Theano 上的 Keras 中构建的。 我正在使用 Pickle 存储它以备将来使用,可能在另一台计算机上。
是否可以在没有 GPU 的情况下使用模型进行预测?
【问题讨论】:
标签: neural-network gpu theano pickle keras
当然。它甚至是一个常见的用例。 GPU 有助于促进训练,但有时在生产中不可用(例如,如果您在客户的手机上运行)。
我不太了解 Theano,但他们可能有一个等价于 tensorflow.serving。您始终可以序列化经过训练的模型对象并从另一台机器读取它。
要序列化,您可以使用:
keras.models.save_model 和 keras.models.load_model 将模型转储到 hdf5 文件。【讨论】: