Update DDP example in README.md (#6384)

pytorch · Apr 22, 2024 · ec3d137 · ec3d137
1 parent ec4b301
commit ec3d137
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/README.md b/README.md
@@ -78,7 +78,7 @@ If you're using `DistributedDataParallel`, make the following changes:
 +import torch_xla.distributed.xla_multiprocessing as xmp
 +import torch_xla.distributed.xla_backend
 
- def _mp_fn(rank, world_size):
+ def _mp_fn(rank):
    ...
 
 -  os.environ['MASTER_ADDR'] = 'localhost'