TensorFlow 2.2 in SageMaker not saving output model for deployment

0

I am experiencing an issue when using TensorFlow version 2.2 in Amazon SageMaker to train my model. The output model is not being saved and cannot be deployed (no model artifact in /output folder). However, when using TensorFlow version 1.15, the model is trained and saved successfully. Can anyone provide insight or a solution for this issue? I am using : framework_version='2.2.0' py_version="py37"

Thank you in advance

alioo
질문됨 2년 전394회 조회
1개 답변
1

There could be multiple reasons for the issue you're facing:

  • Incorrect code: Make sure you've correctly saved the model using tf.saved_model.save or tf.keras.models.save_model functions.
  • Incompatible versions: TensorFlow 2.2 may have compatibility issues with the Amazon SageMaker environment you're using. Try using a different version of TensorFlow or a different environment.
  • Missing dependencies: Make sure you've installed all required dependencies for TensorFlow 2.2.
  • Disk space: Check if the disk space is sufficient for saving the model.
  • Permission issues: Ensure that the user running the training script has the necessary permissions to write to the /output folder.

Try these potential solutions and see if any of them resolve your issue.

profile picture
답변함 2년 전

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

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

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