tests/nn/data_parallel/test_sharded_ddp.py · 6e7ad7989c129f896a679436694dcce2dc34104a · OpenDAS / fairscale · GitLab

Find file Blame History Permalink

[ShardedDDP] Sync buffers + small cleanup (#112) · 79ded821
Benjamin Lefaudeux authored Sep 28, 2020
```
- adding the buffer broadcast option
- minor cleanup in shardedDDP
```
79ded821

test_sharded_ddp.py 3.35 KB