Skip to content

[Bugfix] Fix batch size in slapo-deepspeed#7

Merged
zarzen merged 2 commits intoawslabs:mainfrom
chhzh123:fix_ds_bs
Jan 19, 2023
Merged

[Bugfix] Fix batch size in slapo-deepspeed#7
zarzen merged 2 commits intoawslabs:mainfrom
chhzh123:fix_ds_bs

Conversation

@chhzh123
Copy link
Contributor

@chhzh123 chhzh123 commented Jan 19, 2023

Description

The default micro-batch size at the beginning of the function leads to incorrect arguments passed into the DeepSpeed engine. This PR removes the default micro-batch size.

Checklist

  • PR's title starts with a category (e.g. [Bugfix], [Model], [Tutorial], etc)
  • Changes are complete (i.e. I finished coding on this PR)
  • All changes have test coverage
  • Code is well-documented

@chhzh123 chhzh123 requested a review from comaniac January 19, 2023 10:24
Copy link
Contributor

@comaniac comaniac left a comment

Choose a reason for hiding this comment

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

LGTM
cc @zarzen

@zarzen
Copy link
Contributor

zarzen commented Jan 19, 2023

LGTM

@zarzen zarzen merged commit 353d6a7 into awslabs:main Jan 19, 2023
chhzh123 added a commit to chhzh123/slapo that referenced this pull request Jan 22, 2023
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