PR #5936 Fix zero-1 bug for inferring local ranks

Fix zero-1 bug for inferring local ranks #5936

alanwaketan merged 2 commits into pytorch:master from YangFei1990:zero_1_local_rank_fix

fix zero-1 bug for inferring local ranks

f0dd2a16

formatting

cc417a85

JackCaoG requested a review from

alanwaketan 2 years ago

JackCaoG added fsdp

JackCaoG added backport_2.2

alanwaketan approved these changes on 2023-12-14

alanwaketan merged 5577dd78 into master 2 years ago

Reviewers

alanwaketan

Assignees

No one assigned

Labels

fsdp backport_2.2

Milestone

No milestone