无法在Turicreate中导出Core ML模型

时间:2020-05-16 19:44:49

标签: python-3.x jupyter-notebook amazon-sagemaker coreml turi-create

我将AWS Sagemaker与Jupyter笔记本一起使用来训练Turicreate模型。它训练成功,但是我无法将其导出到CoreML模型。它显示以下错误。我已经在Jupyter笔记本中尝试了各种内核,结果相同。关于如何解决此错误的任何想法?

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

0 个答案:

没有答案