PR #7252 [ZeRO-1] Only warn on rank-0

[ZeRO-1] Only warn on rank-0 #7252

hgt312 wants to merge 1 commit into r2.1_aws_neuron from hgt312/zero1_warn_patch

only warn on rank-0

1d3e9a7e

JackCaoG approved these changes on 2024-06-12

Reviewers

JackCaoG

Assignees

No one assigned

Labels

None yet

Milestone

No milestone