我已将Keras模型转换为CoreML。我想确保CoreML使用半浮动而不是所有纹理的完整浮动。我怎样才能做到这一点?
更新: 如何确保网络的输出是半浮点数或至少是Float(或任何其他类型)而不是Double?
答案 0 :(得分:2)
如果在GPU上运行CoreML模型,则无需执行任何操作。 MPS(金属性能着色器)将自动使用半浮动。如果在CoreML模型上运行GPU Frame Capture,则可以看到这一点。
我写了一篇关于CoreML如何在幕后工作的博客文章,它实际上证明了它使用了半浮动:http://machinethink.net/blog/peek-inside-coreml/