首页 > 解决方案 > 包含自定义层的 Core ML 模型很慢

问题描述

我已将 coremltools(caffe 更改为 Core ML,https: //apple.github.io/coremltools/generated/coremltools.converters.caffe.convert.html#coremltools.converters.caffe.convert )并将自定义层添加到 Core ML模型。但与非自定义 caffe 层 Core ML 模型相比,Core ML(包括自定义层)速度较慢。我发现coremlc编译了mlmodel,并且融合了一些层,例如conv+bn+scale,conv+relu。 conv+bn+scale 融合conv+relu 融合

我用instrument(time profiler)做Core ML模型,发现函数(time-consumed)(自定义层Core ML模型)比(非自定义层Core ML模型)多,大家知道为什么吗? CPU 上的 Core ML 模型时间分析器

标签: caffecoreml

解决方案


推荐阅读