[bug] hotfixes to OSS Readme (#94)

facebookresearch · Sep 17, 2020 · 656828b · 656828b
1 parent 2d415f3
commit 656828b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -21,10 +21,11 @@ model = fairscale.nn.Pipe(model, balance=[2, 2], devices=[0, 1], chunks=8)
 ```
 
 ### Optimizer state sharding (ZeRO)
-See a more complete example [here](https://github.com/facebookresearch/fairscale/blob/oss_async_broadcast/benchmarks/oss.py), but a minimal example could look like the following :
+See a more complete example [here](https://github.com/facebookresearch/fairscale/blob/master/benchmarks/oss.py), but a minimal example could look like the following :
 
 ```python
 import torch
+import torch.multiprocessing as mp
 from fairscale.optim.oss import OSS
 
 def train(