Commits · bc82290241be33c0159c641dc518f5d095b03f3e · OpenDAS / fairscale · GitLab

13 Aug, 2020 4 commits
- [refactor] remove type_shim.h (#33) · bc822902
  Jun Ru Anderson authored Aug 13, 2020
  
  bc822902
- [chore] run tests on PyTorch 1.6.0 and gpu tests on 1.6.0 and 1.5.1 (#34) · 571f5efa
  msbaines authored Aug 13, 2020
  
  571f5efa
- [feat] remove support for non-multitensor Adam · 81a2cf04
  Jun Ru Anderson authored Aug 13, 2020
```
Co-authored-by: Jun Ru Anderson <andersonic@fb.com>
```
  81a2cf04
- Aligning OSS state dict with... · 57079b08
  Benjamin Lefaudeux authored Aug 12, 2020
```
Aligning OSS state dict with `https://pytorch.org/docs/stable/_modules/torch/optim/optimizer.html#Optimizer` (#31)
```
  57079b08
08 Aug, 2020 1 commit
- [fix] fix test_oss.py when host have 2 GPUs (#26) · d9e6ceaa
  Min Xu authored Aug 07, 2020
```
Co-authored-by: Min Xu <m1n@fb.com>
```
  d9e6ceaa
06 Aug, 2020 2 commits
- [feat] add ddp that works with oss with reduce() not all_reduce() (#19) · 525e709b
  Min Xu authored Aug 06, 2020
```
Co-authored-by: Min Xu <m1n@fb.com>
```
  525e709b
- add pytest coverage (#24) · 4cd2590c
  Min Xu authored Aug 06, 2020
```
Co-authored-by: Min Xu <m1n@fb.com>
```
  4cd2590c
01 Aug, 2020 1 commit
- [cleanup] fix setup.py description (#22) · 7f3c07bb
  msbaines authored Jul 31, 2020
  
  7f3c07bb
31 Jul, 2020 11 commits
- [fix] add build dependency on torch (#21) · 291da1ea
  msbaines authored Jul 31, 2020
  
  291da1ea
- [chore] setup a proper version number (#20) · 5a2b5720
  msbaines authored Jul 31, 2020
  
  5a2b5720
- [feat] Implement OSS save and load of the sharded state from a single replica (#16) · 8e363567
  Benjamin Lefaudeux authored Jul 31, 2020
  
  8e363567
- [feat] add FusedAdam (#10) · bfba68d8
  Jun Ru Anderson authored Jul 30, 2020
```
Add FusedAdam, update benchmark and add tests.
Co-authored-by: Jun Ru Anderson <andersonic@fb.com>
```
  bfba68d8
- [doc] added a bit details to initial setup (#14) · 4f7d7d34
  Min Xu authored Jul 30, 2020
  
  4f7d7d34
- [doc] Fix copyright in stubs (#13) · e10de4b5
  msbaines authored Jul 30, 2020
  
  e10de4b5
- [feat] Model parallel (#3) · 30f5009a
  Tom Birch authored Jul 22, 2020
  
  30f5009a
- [test] Use PyTorch v1.5 for ci (#7) · 8634280c
  msbaines authored Jul 22, 2020
  
  8634280c
- [fix] add TransformerEncoderLayer to stubs (#5) · 63b5b166
  Jun Ru Anderson authored Jul 21, 2020
```
Co-authored-by: Jun Ru Anderson <andersonic@fb.com>
```
  63b5b166
- [test] switch benchmark to use Adam rather than SGD · e4b2ffd4
  Jun Ru Anderson authored Jul 17, 2020
  
  e4b2ffd4
- [feat] add Transformer gpipe benchmark · 74181b08
  Jun Ru Anderson authored Jul 17, 2020
  
  74181b08
08 Jul, 2020 1 commit
- Initial commit · 0cd65242
  Mandeep Singh Baines authored Jul 07, 2020
  
  0cd65242