CoreML初始化失敗：MTLTextureDescriptor的寬度大於最大允許大小

Question

我有一個Keras CNN模型，已使用coremltools轉換為CoreML 。 它可以完美地在模擬器上運行，但不能在iPhone X上運行。僅在初始化時崩潰：

let cnn = MyCnn()

…有錯誤：

MTLTextureDescriptor的寬度（44100）大於最大允許大小16384。

閱讀了類似錯誤的SO帖子后，我認為這可能是GPU問題，因此我嘗試設置

usesCPUOnly = true 

但是，我只能在預測期間發送選項，並且似乎無法在初始化之前進行設置（發生崩潰的位置）。

有什么想法我能做什么？

Answer 1

您可以在初始化期間通過傳遞MLModelConfiguration對象來設置它。

但是，使模型更小可能是一個好主意。 聽起來這對於手機來說太大了。