Uh oh!
There was an error while loading. Please reload this page.
local mode: support output_path - #449
Merged
Merged
Conversation
Can be either file:// or s3:// - This also changes the default behavior of local mode to use the SDK provided default S3 bucket if nothing is passed. This makes it easier for customers to create models in SageMaker too since their Model Artifacts will already be a tarfile in S3.
codecov-io
commented
Oct 29, 2018
Codecov Report
@@ Coverage Diff @@## master #449 +/- ##
==========================================
- Coverage 93.75% 93.68% -0.08%
==========================================
Files 55 55 Lines 4034 4051 +17 ==========================================
+ Hits 3782 3795 +13 - Misses 252 256 +4
Continue to review full report at Codecov.
|
nadiaya
reviewed
Oct 30, 2018
| self._current_job_name = None | ||
| if (not self.sagemaker_session.local_mode | ||
| and output_path and output_path.startswith('file://')): | ||
| raise RuntimeError('file:// output paths are only supported in Local Mode') |
| model_files = [os.path.join(model_artifacts, name) for name in os.listdir(model_artifacts)] | ||
| output_files = [os.path.join(output_artifacts, name) for name in os.listdir(output_artifacts)] | ||
| sagemaker.utils.create_tar_file(model_files, os.path.join(compressed_artifacts, 'model.tar.gz')) | ||
| sagemaker.utils.create_tar_file(output_files, os.path.join(compressed_artifacts, 'output.tar.gz')) |
Contributor
There was a problem hiding this comment.
Should that be a function? (all 5 lines)
that seems a bit repetitive.
| _delete_tree(output_artifacts) | ||
| return s3_model_artifacts | ||
| return os.path.join(output_data, 'model.tar.gz') |
Contributor
There was a problem hiding this comment.
Since we moved both files should we return both (model and output) too??
ContributorAuthor
There was a problem hiding this comment.
Not really, SageMaker just returns the S3ModelArtifacts - if you want to look at output.tar.gz you basically have to do a replace in the string. This gets sent directly to the local client describeTrainingJob()
nadiaya
previously approved these changes
Nov 2, 2018
nadiaya
approved these changes
Nov 2, 2018
Evan-W-ang added a commit
to Evan-W-ang/sagemaker-python-sdk
that referenced
this pull request
Jun 8, 2026
output_path is now honored: Can be either file:// or s3:// - This also changes the default behavior of local mode to use the SDK provided default S3 bucket if nothing is passed. This makes it easier for customers to create models in SageMaker too since their Model Artifacts will already be a tarfile in S3. input_channel content_type is now honored in the same way as SageMaker. If it is not provided it is not passed to the container. Before we were always passing 'application/octet-stream'
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for freeto join this conversation on GitHub.
Already have an account?
Sign in to comment
3 participants
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Can be either file:// or s3:// - This also changes the default
behavior of local mode to use the SDK provided default S3 bucket
if nothing is passed. This makes it easier for customers to create
models in SageMaker too since their Model Artifacts will already be
a tarfile in S3.
Issue #, if available:
Description of changes:
Merge Checklist
Put an
xin the boxes that apply. You can also fill these out after creating the PR. If you're unsure about any of them, don't hesitate to ask. We're here to help! This is simply a reminder of what we are going to look for before merging your pull request.By submitting this pull request, I confirm that my contribution is made under the terms of the Apache 2.0 license.