CoreML初始化失败:MTLTextureDescriptor的宽度大于允许的最大大小

时间:2019-02-09 21:16:11

标签: ios coreml coremltools

我有一个Keras CNN模型,我使用coremltools将其转换为CoreML。它可以完美地在模拟器上运行,但不能在iPhone X上运行。仅在初始化时崩溃:

let cnn = MyCnn()

…有错误:

  

MTLTextureDescriptor的宽度(44100)大于最大允许大小16384。

阅读了类似错误的SO帖子后,我认为这可能是GPU问题,因此我尝试设置

usesCPUOnly = true 

但是,我只能在预测期间发送选项,并且似乎无法在初始化(崩溃发生的位置)之前进行设置。

有什么想法我能做什么?

1 个答案:

答案 0 :(得分:1)

您可以在初始化期间通过传递MLModelConfiguration对象来对其进行设置。

但是,使模型更小可能是一个好主意。听起来这对于手机来说太大了。