Add gradient accumulation to SMP v2 train loop

aws / amazon-sagemaker-examples

Example 📓 Jupyter notebooks that demonstrate how to build, train, and deploy machine learning models using 🧠 Amazon SageMaker.

https://sagemaker-examples.readthedocs.io

Apache License 2.0

9.79k stars 6.67k forks source link

Add gradient accumulation to SMP v2 train loop #4596

Open haozhx23 opened 3 months ago

haozhx23 commented 3 months ago

Issue #, if available: Add new function.

Description of changes: Add a straight-forward gradient accumulation function to the train_step and arguments.py

Testing done: Behave as expected.

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.

[x] I have read the CONTRIBUTING doc and adhered to the example notebook best practices
[ ] I have updated any necessary documentation, including READMEs
[x] I have tested my notebook(s) and ensured it runs end-to-end
[x] I have linted my notebook(s) and code using black-nb -l 100 {path}/{notebook-name}.ipynb

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

sagemaker-bot commented 3 months ago

AWS CodeBuild CI Report

CodeBuild project: SageMakerNotebooksProd-USWEST2-amazon-sagemaker-examples-pr
Commit ID: 35e519390bf2cb5f6f48c0010c8355af3947ed04
Result: SUCCEEDED
Build Logs (available for 30 days)

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

sagemaker-bot commented 3 months ago

AWS CodeBuild CI Report

CodeBuild project: SageMakerNotebooksProd-USWEST2-sagemaker-examples-grammar
Commit ID: 35e519390bf2cb5f6f48c0010c8355af3947ed04
Result: SUCCEEDED
Build Logs (available for 30 days)

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

sagemaker-bot commented 3 months ago

AWS CodeBuild CI Report

CodeBuild project: SageMakerNotebooksProd-USWEST2-sagemaker-examples-link-check
Commit ID: 35e519390bf2cb5f6f48c0010c8355af3947ed04
Result: SUCCEEDED
Build Logs (available for 30 days)

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

sagemaker-bot commented 3 months ago

AWS CodeBuild CI Report

CodeBuild project: SageMakerNotebooksProd-USWEST2-sagemaker-examples-code-formatting
Commit ID: 35e519390bf2cb5f6f48c0010c8355af3947ed04
Result: SUCCEEDED
Build Logs (available for 30 days)

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