SageMaker Neo Compilation Input Error - Converting Tensflor to CoreML

0

Hi all,

I have a model that I've finetuned from the TF MobileResnet model available in SageMaker. The model is working well, and returns correct inference values when I host it using the SageMaker inference endpoints.

I am now trying to convert the model to coreml, however when attempting to run this job via SageMaker Neo I am receiving the following error:

ClientError: InputConfiguration: Unable to determine the type of the model, i.e. the source framework. Please provide the value of argument "source", from one of ["tensorflow", "pytorch", "mil"]. Note that model conversion requires the source package that generates the model. Please make sure you have the appropriate version of source package installed. E.g., if you\'re converting model originally trained with TensorFlow 1.14, make sure you have `tensorflow==1.14` installed. tensorflow 2.2.0, coremltools 4.0 For further troubleshooting common failures please visit: https://docs.aws.amazon.com/sagemaker/latest/dg/neo-troubleshooting-compilation.html

I cannot find any information about this error in the docs linked in the error message. Any help would be greatly appreciated.

질문됨 일 년 전78회 조회
답변 없음

로그인하지 않았습니다. 로그인해야 답변을 게시할 수 있습니다.

좋은 답변은 질문에 명확하게 답하고 건설적인 피드백을 제공하며 질문자의 전문적인 성장을 장려합니다.

질문 답변하기에 대한 가이드라인

관련 콘텐츠