xla
294610a1 - Fix the missing parameter error when running mp_imagenet with torchrun (#5729)

Commit

1 year ago

Fix the missing parameter error when running mp_imagenet with torchrun (#5729) * Fix the missing parameter error when running mp_imagenet with torchrun * made it local rank

References

#5729 - Fix the missing parameter error when running mp_imagenet with torchrun

Author

vanbasten23

Parents

ecc0e230

xla 294610a1 - Fix the missing parameter error when running mp_imagenet with torchrun (#5729)

xla
294610a1 - Fix the missing parameter error when running mp_imagenet with torchrun (#5729)