我使用 AWS Sagemaker 和 Jupyter 笔记本来训练我的 Turicreate 模型。它训练成功,但我无法将其导出到 CoreML 模型。它显示以下错误。我在 Jupyter notebook 中尝试了各种内核,结果相同。有关如何解决此错误的任何想法?
turicreate 5.4 GPU:mxnet-cu100
KeyError Traceback (most recent call last)
<ipython-input-6-3499bdb76e06> in <module>()
1 # Export for use in Core ML
----> 2 model.export_coreml('pushupsTC.mlmodel')
~/anaconda3/envs/python3/lib/python3.6/site-packages/turicreate/toolkits/object_detector/object_detector.py in export_coreml(self, filename, include_non_maximum_suppression, iou_threshold, confidence_threshold)
1216 assert (self._model[23].name == 'pool5' and
1217 self._model[24].name == 'specialcrop5')
-> 1218 del net._children[24]
1219 net._children[23] = op
1220
KeyError: 24