我有一个要求,我需要在脱机设备中部署卷积神经网络模型。我知道我们可以使用Google Cloud ML来训练模型,调整超超参数并将其部署用于预测。
但我的问题是我们是否可以下载经过培训的TensorFlow模型并将其部署在自定义设备上进行预测?
注意 - 自定义设备将具有大量处理能力,但没有互联网连接。
答案 0 :(得分:3)
是。培训服务和预测服务完全分开。要为自定义设备训练模型,请创建TensorFlow脚本以训练模型。该脚本通常将描述一个用于训练的TensorFlow图和用于预测的第二个图(预测图将以可以加载训练期间学习的参数的方式构造)。预测图将根据您的自定义硬件进行定制。
确保包含将预测图导出到GCS的命令。有关如何导出模型的示例,请参阅此post。