我有一个Keras CNN模型,我使用coremltools将其转换为CoreML。它可以完美地在模拟器上运行,但不能在iPhone X上运行。仅在初始化时崩溃:
let cnn = MyCnn()
…有错误:
MTLTextureDescriptor的宽度(44100)大于最大允许大小16384。
阅读了类似错误的SO帖子后,我认为这可能是GPU问题,因此我尝试设置
usesCPUOnly = true
但是,我只能在预测期间发送选项,并且似乎无法在初始化(崩溃发生的位置)之前进行设置。
有什么想法我能做什么?
答案 0 :(得分:1)
您可以在初始化期间通过传递MLModelConfiguration
对象来对其进行设置。
但是,使模型更小可能是一个好主意。听起来这对于手机来说太大了。