Сбой инициализации CoreML: ширина MTLTextureDescriptor превышает максимально допустимый размер

Question

У меня есть модель Keras CNN, которую я преобразовал в CoreML, используя coremltools .Он отлично работает на симуляторе, но не на iPhone X. Он падает только при инициализации:

let cnn = MyCnn()

… с ошибкой:

MTLTextureDescriptor имеет ширину (44100) больше максимальнойдопустимый размер 16384.

После прочтения SO сообщений с похожими ошибками я подумал, что, возможно, это проблема с графическим процессором, поэтому я попытался установить

usesCPUOnly = true 

Однако яможет отправлять параметры только во время прогнозирования, и, кажется, нет способа установить это до инициализации - там, где происходит сбой.

Есть идеи, что я могу сделать?

Answer 1

Вы можете установить его во время инициализации, передав объект MLModelConfiguration.

Но, вероятно, хорошая идея сделать вашу модель меньше.Похоже, это слишком много для мобильного телефона.