Skip to content

Add model saving warning at end of training #171

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 4 commits into from
Feb 28, 2019

Conversation

icywang86rui
Copy link
Contributor

  • Add warning when no model artifact is found
  • Add warning if model is not saved in the SavedModel bundle format

Issue #, if available:

Description of changes:

By submitting this pull request, I confirm that my contribution is made under the terms of the Apache 2.0 license.

* Add warning when no model artifact is found
* Add warning if model is not saved in the SavedModel bundle format
.format(direct_parent_dir))

if not file_exists:
logger.warn('No model artifact is saved under path {}.'
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Shouldn't we toss an exception here?

Are there any use cases of users not utilizing /opt/ml/model for packaging their models to S3?

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

They could have saved models or checkpoints to s3 during training. saving it to '/opt/ml/model' just to make sure ease will upload it at end of training.

Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Can you provide any links or references to users expecting to have had their training artifacts in /opt/ml/model, however they didn't know?

@icywang86rui icywang86rui merged commit 686ae25 into aws:script-mode Feb 28, 2019
Elizaaaaa pushed a commit to Elizaaaaa/sagemaker-tensorflow-container that referenced this pull request Nov 4, 2019
* Add model saving warning at end of training

* Add warning when no model artifact is found
* Add warning if model is not saved in the SavedModel bundle format

* Combine logging messages

* Enforce psutil version

* Remove pinned version of sagemaker-containers and install this package last
Elizaaaaa pushed a commit to Elizaaaaa/sagemaker-tensorflow-container that referenced this pull request Nov 4, 2019
* Add model saving warning at end of training

* Add warning when no model artifact is found
* Add warning if model is not saved in the SavedModel bundle format

* Combine logging messages

* Enforce psutil version

* Remove pinned version of sagemaker-containers and install this package last
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants