-
Notifications
You must be signed in to change notification settings - Fork 1.1k
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
breaking: remove estimator parameters for TF legacy mode #1510
Conversation
AWS CodeBuild CI Report
Powered by github-codebuild-logs, available on the AWS Serverless Application Repository |
AWS CodeBuild CI Report
Powered by github-codebuild-logs, available on the AWS Serverless Application Repository |
AWS CodeBuild CI Report
Powered by github-codebuild-logs, available on the AWS Serverless Application Repository |
AWS CodeBuild CI Report
Powered by github-codebuild-logs, available on the AWS Serverless Application Repository |
AWS CodeBuild CI Report
Powered by github-codebuild-logs, available on the AWS Serverless Application Repository |
AWS CodeBuild CI Report
Powered by github-codebuild-logs, available on the AWS Serverless Application Repository |
AWS CodeBuild CI Report
Powered by github-codebuild-logs, available on the AWS Serverless Application Repository |
AWS CodeBuild CI Report
Powered by github-codebuild-logs, available on the AWS Serverless Application Repository |
class TensorFlow(Framework): | ||
"""Handle end-to-end training and deployment of user-provided TensorFlow code.""" | ||
|
||
__framework_name__ = "tensorflow" | ||
_SCRIPT_MODE_REPO_NAME = "tensorflow-scriptmode" |
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.
Is this repo name used by the DLC images?
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.
no, it's used only by the TF 1.11-1.13 images. this PR still preserves the image URI stuff for older images, though arguably it could be removed as well. definitely open to it.
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.
Makes sense. I think it's probably safe to remove that functionality for older images and require users to specify the image URI directly, but I'm also fine with leaving it as is.
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.
I'll probably do it in a separate PR then, and add some appropriate warnings. maybe once #1464 is finished, it'll be cleaner.
Issue #, if available:
#1462
Description of changes:
This change removes a lot of the "training helpers" associated with TF legacy mode, such as:
training_steps
,evaluation_steps
,checkpoint_path
, andrequirements_file
, which currently are simply translated into hyperparametersstill left:
framework_version
< 1.11 to specify the image URI directlyTesting done:
tox --parallel all tests/unit
Merge Checklist
Put an
x
in 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.General
Tests
unique_name_from_base
to create resource names in integ tests (if appropriate)By submitting this pull request, I confirm that my contribution is made under the terms of the Apache 2.0 license.