tests/experimental/nn/test_auto_shard.py · 86c62cc9c989178e629823c2fd4a3cc11723e92b · OpenDAS / fairscale · GitLab

Find file Blame History Permalink

Use correct node names for param counting in auto_shard. (#830) · 86c62cc9
Eugen Hotaj authored Oct 26, 2021
```
Fixes #827.
Co-authored-by: Eugen Hotaj <ehotaj@fb.com>
```
86c62cc9

test_auto_shard.py 4.97 KB