Skip to content

[SageMaker Data Parallel] Upgrade the TF2 version to 2.4.1 and also update the py_version to 3.7 #2069

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 3 commits into from
Mar 10, 2021
Merged

[SageMaker Data Parallel] Upgrade the TF2 version to 2.4.1 and also update the py_version to 3.7 #2069

merged 3 commits into from
Mar 10, 2021

Conversation

karan6181
Copy link
Contributor

Description of changes:

  • Upgraded the SageMaker Data Parallel TensorFlow2 version to 2.4.1

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.

@review-notebook-app
Copy link

Check out this pull request on  ReviewNB

See visual diffs & provide feedback on Jupyter Notebooks.


Powered by ReviewNB

@sagemaker-bot
Copy link
Collaborator

AWS CodeBuild CI Report

  • CodeBuild project: amazon-sagemaker-examples-pr
  • Commit ID: ee10667
  • Result: FAILED
  • Build Logs (available for 30 days)

Powered by github-codebuild-logs, available on the AWS Serverless Application Repository

@karan6181
Copy link
Contributor Author

karan6181 commented Mar 10, 2021

Attached here is TF2 MNIST and Mask R-CNN jupyter notebook (in a zip file) with the output by attaching FSx Lustre
tensorflow2_smdataparallel_mnist_maskrcnn_demo.zip. Both the test passes successfully. I have removed the subnet, security group, and FSx file system ID due to security issues.

Copy link
Contributor

@ChoiByungWook ChoiByungWook left a comment

Choose a reason for hiding this comment

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

Awesome! LGTM! I checked the zip file and can confirm the notebook seems to run as expected! Overriding and merging, similar to the most recent previous PR that updated this notebook: #1882

@ChoiByungWook ChoiByungWook merged commit a2bf7cf into aws:master Mar 10, 2021
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