如何使用带有CoreML神经网络的半浮动?

时间:2017-08-30 18:35:17

标签: coreml

我已将Keras模型转换为CoreML。我想确保CoreML使用半浮动而不是所有纹理的完整浮动。我怎样才能做到这一点?

更新: 如何确保网络的输出是半浮点数或至少是Float(或任何其他类型)而不是Double?

1 个答案:

答案 0 :(得分:2)

如果在GPU上运行CoreML模型,则无需执行任何操作。 MPS(金属性能着色器)将自动使用半浮动。如果在CoreML模型上运行GPU Frame Capture,则可以看到这一点。

我写了一篇关于CoreML如何在幕后工作的博客文章,它实际上证明了它使用了半浮动:http://machinethink.net/blog/peek-inside-coreml/