Commits · d17a015f9bc3099e90e74a41ec1584a6fd76e14f · OpenDAS / apex · GitLab

25 May, 2018 1 commit
- Transferred backend and build system to use Pytorch C++ extension + ATen dispatch. · d17a015f
  Michael Carilli authored May 25, 2018
  
  d17a015f
18 May, 2018 1 commit
- Initial support for automatic mixed precision · e733e78c
  Carl Case authored May 16, 2018
  
  e733e78c
17 May, 2018 1 commit
- Merge pull request #1 from raulpuric/patch-1 · a3059288
  Christian Sarofeen authored May 17, 2018
```
mLSTM cpu fix
```
  a3059288
16 May, 2018 1 commit
- Adding minimal examples with Apex and Pytorch distributed data parallel · 8e1f01c5
  Michael Carilli authored May 16, 2018
  
  8e1f01c5
15 May, 2018 2 commits
- mLSTM cpu fix · 435133f1
  Raul Puri authored May 15, 2018
  
  435133f1
- Fix build for pytorch post 0.4 · 83acda92
  Christian Sarofeen authored May 15, 2018
  
  83acda92
14 May, 2018 4 commits
- Multi-op sequence for ddp_race_condition_test.py · cc8f03c8
  Michael Carilli authored May 14, 2018
  
  cc8f03c8
- Revert distributed to simpler version until backward hooks are fixed. · 47ac5c2b
  Christian Sarofeen authored May 14, 2018
  
  47ac5c2b
- Manually setting rank in imagenet examples · d4066f6e
  Michael Carilli authored May 14, 2018
  
  d4066f6e
- Updating distributed example · 9661dbd7
  Michael Carilli authored May 14, 2018
  
  9661dbd7
09 May, 2018 1 commit
- Need to set_epoch in distributed example for sampler · 789afd89
  Christian Sarofeen authored May 09, 2018
  
  789afd89
08 May, 2018 1 commit
- Remove stub sentence in doc · 3ab105d9
  Christian Sarofeen authored May 08, 2018
  
  3ab105d9
07 May, 2018 2 commits
- Fix race condition in DDP. · 7c2ae41e
  Christian Sarofeen authored May 01, 2018
  
  7c2ae41e
- Updating word_language_model examples to evaluate with no_grad instead of volatile · 531cb5c3
  Michael Carilli authored May 07, 2018
  
  531cb5c3
06 May, 2018 1 commit
- Adding main_fp16_optimizer.py to examples/imagenet and word_language_model · 0d91a65e
  Michael Carilli authored May 05, 2018
  
  0d91a65e
04 May, 2018 2 commits
- Don't need libcuda for building. · 9b76c2a2
  Christian Sarofeen authored May 04, 2018
  
  9b76c2a2
- Typo · 2c1cc0b5
  Christian Sarofeen authored May 04, 2018
  
  2c1cc0b5
03 May, 2018 1 commit
- Adding examples that can serve as master copies for QA workflow moving forward. · 8ef50822
  Michael Carilli authored May 03, 2018
  
  8ef50822
02 May, 2018 2 commits
- Editing setup.py to locate nvcc and detect cuda major version more strictly · 5dfa4c37
  Michael Carilli authored May 02, 2018
  
  5dfa4c37
- Changing promotion of FP16_optimizer.optimizer attributes to use properties · b0d7d60d
  Michael Carilli authored May 01, 2018
  
  b0d7d60d
01 May, 2018 4 commits
- Try to improve robustness of finding cuda in build. Try to support building with CUDA 8. · 5709cfb5
  Christian Sarofeen authored May 01, 2018
  
  5709cfb5
- Don't compile cuda code if running clean. · 1cea1005
  Christian Sarofeen authored May 01, 2018
  
  1cea1005
- Distributed refactor. · 3a08d827
  Christian Sarofeen authored May 01, 2018
  
  3a08d827
- properly overriding __getstate__ and __setstate__ · dd05dbdb
  Michael Carilli authored Apr 30, 2018
  
  dd05dbdb
30 Apr, 2018 1 commit
- Reorganizing fp16_optimizer · 7ce2e04f
  Michael Carilli authored Apr 27, 2018
  
  7ce2e04f
26 Apr, 2018 3 commits
- Updating documentation · 31cee8e7
  Michael Carilli authored Apr 26, 2018
  
  31cee8e7
- Adding more helpful exception messages to prep_param_lists · 262da9c6
  Michael Carilli authored Apr 26, 2018
  
  262da9c6
- step() now forwards return value · f22201e7
  Michael Carilli authored Apr 25, 2018
  
  f22201e7
25 Apr, 2018 5 commits
- Cleaned comments in fp16_utils and csrc. Keeping comments that are non-docstring but informative. · a3e2776a
  Michael Carilli authored Apr 25, 2018
  
  a3e2776a
- FP16 util fix, detach at the right point. · a288d585
  Christian Sarofeen authored Apr 25, 2018
  
  a288d585
- Update README.md · 65167d21
  Christian Sarofeen authored Apr 25, 2018
  
  65167d21
- Distributed fix for non-all reduce call. · 866454e0
  Christian Sarofeen authored Apr 25, 2018
  
  866454e0
- Initial release · 2fa4dbaf
  Christian Sarofeen authored Apr 25, 2018
  
  2fa4dbaf