我有一个要求,需要在离线设备中部署卷积神经网络模型。我知道我们可以使用Google Cloud ML训练模型,调整超参数并将其部署用于预测。
但我的问题是我们是否可以下载经过训练的 TensorFlow 模型并将其部署在自定义设备上进行预测?
注意 - 自定义设备将具有强大的处理能力,但没有互联网连接。
最佳答案
是的。训练服务和预测服务是完全分开的。要为自定义设备训练模型,您可以创建一个 TensorFlow 脚本来训练模型。该脚本通常会描述一个用于训练的 TensorFlow 图和另一个用于预测的 TensorFlow 图(预测图的构建方式将使其能够加载训练期间学到的参数)。预测图将为您的定制硬件量身定制。
请务必包含将预测图导出到 GCS 的命令。有关如何导出模型的示例,请参阅此 post .
关于tensorflow - 我可以从 Google Cloud ML Engine 下载训练模型吗?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/44400546/