-
Notifications
You must be signed in to change notification settings - Fork 162
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
Conversation
* 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 {}.' |
There was a problem hiding this comment.
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?
There was a problem hiding this comment.
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.
There was a problem hiding this comment.
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?
* 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
* 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
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.