Sagemaker Neo Compilation error for Tensorflow

aws / amazon-sagemaker-examples

Example 📓 Jupyter notebooks that demonstrate how to build, train, and deploy machine learning models using 🧠 Amazon SageMaker.

https://sagemaker-examples.readthedocs.io

Apache License 2.0

10.14k stars 6.78k forks source link

Sagemaker Neo Compilation error for Tensorflow #1322

Open subunithya opened 4 years ago

subunithya commented 4 years ago

When I try to compile the Faster-RCNN-Inception model which was taken from Model Zoo TF-V1. I am getting the following error:

ClientError: InputConfiguration: Framework cannot load Tensorflow model: as_list() is not defined on an unknown TensorShape.

How to resolve this issue?

ivansmith8895 commented 4 years ago

Seeing this error as well on jobs that previously worked via compilations. Can someone investigate? My input config conforms to the documented format available below.

https://docs.aws.amazon.com/sagemaker/latest/dg/neo-job-compilation-cli.html