1. 01 Dec, 2020 1 commit
  2. 30 Nov, 2020 1 commit
  3. 27 Nov, 2020 1 commit
  4. 26 Nov, 2020 1 commit
  5. 24 Nov, 2020 1 commit
  6. 22 Nov, 2020 1 commit
  7. 21 Nov, 2020 1 commit
    • Benjamin Lefaudeux's avatar
      [feat] ShardedDataParallel with autoreduce (#157) · ad933b34
      Benjamin Lefaudeux authored
      * rewrite using autograd and Variable execution queue to make the reduce automatic
      * share buckets with OSS to remove duplication
      * some speed still likely on the table since the speed vs. bucketing does not match expectations, could be a follow up
      ad933b34
  8. 20 Nov, 2020 1 commit
  9. 19 Nov, 2020 4 commits
  10. 18 Nov, 2020 2 commits
  11. 17 Nov, 2020 1 commit
    • Min Xu's avatar
      [doc] add AdaScale API doc (#191) · 587b707d
      Min Xu authored
      - removed experimental warning as we have validated it on cifar and
      imagenet, transformer is looking good so far too.
      - fixed API doc formatting
      - make it consistent with the other code in the repo
      - tested by making the doc locally and inspect the results
      587b707d
  12. 16 Nov, 2020 1 commit
  13. 12 Nov, 2020 2 commits
  14. 11 Nov, 2020 2 commits
  15. 10 Nov, 2020 1 commit
    • Tom Birch's avatar
      Single-process control via PipeRPCWrapper (#156) · 5d4f50fb
      Tom Birch authored
      Adds support for:
      * Reused layers (e.g. for weight sharing)
      * Lazily-constructed layers
      * Single-process control via PipeRPCWrapper
      * PipelineStyle.AsyncScheudle, which lays the foundation for asynchronous pipeline work by introducing an event loop for each rank/worker to process either activations or gradients as they arrive
      
      Also added examples for multi-process and PipeRPCWrapper
      5d4f50fb
  16. 06 Nov, 2020 2 commits
  17. 04 Nov, 2020 1 commit
  18. 30 Oct, 2020 2 commits
  19. 29 Oct, 2020 1 commit
  20. 28 Oct, 2020 2 commits
  21. 26 Oct, 2020 1 commit
  22. 23 Oct, 2020 3 commits
  23. 22 Oct, 2020 3 commits
  24. 21 Oct, 2020 4 commits
    • Min Xu's avatar
      [fix] fixing adascale all_reduce (#155) · 6802ad49
      Min Xu authored
      - Aurick noticed this bug and I ran into it yesterday
      - after the fix, our cifar training shows same gain values from
        different replics now:
      
      ```
      20-Oct-20 16:00:19 - DEBUG - rank1 - scale 2, gain ratio 1.3512124098087777
      20-Oct-20 16:00:19 - DEBUG - rank0 - scale 2, gain ratio 1.3512124098087777
      20-Oct-20 16:00:19 - DEBUG - rank1 - timing: data 0:00:00.000600 fwd 0:00:00.003678 loss 0:00:00.000086 bwd 0:00:00.314158 update 0:00:00.002132 rest 0:00:00.000399
      20-Oct-20 16:00:19 - DEBUG - rank0 - timing: data 0:00:00.000643 fwd 0:00:00.003460 loss 0:00:00.000084 bwd 0:00:00.314678 update 0:00:00.002001 rest 0:00:00.000408
      20-Oct-20 16:00:19 - DEBUG - rank1 - scale 2, gain ratio 1.3514997779980324
      20-Oct-20 16:00:19 - DEBUG - rank0 - scale 2, gain ratio 1.3514997779980324
      20-Oct-20 16:00:19 - DEBUG - rank1 - timing: data 0:00:00.000732 fwd 0:00:00.003689 loss 0:00:00.000086 bwd 0:00:00.314176 update 0:00:00.002146 rest 0:00:00.000397
      20-Oct-20 16:00:19 - DEBUG - rank0 - timing: data 0:00:00.000646 fwd 0:00:00.003542 loss 0:00:00.000089 bwd 0:00:00.314549 update 0:00:00.001956 rest 0:00:00.000392
      20-Oct-20 16:00:19 - DEBUG - rank1 - scale 2, gain ratio 1.352149646693932
      20-Oct-20 16:00:19 - DEBUG - rank0 - scale 2, gain ratio 1.352149646693932
      ```
      6802ad49
    • Benjamin Lefaudeux's avatar
      [feature] OSS: Use MNIST to benchmark (#159) · 6f8a8652
      Benjamin Lefaudeux authored
      * switching to MNIST
      * updating the reference values, should be good to go
      * download dataset once for all processes
      6f8a8652
    • Vittorio Caggiano's avatar
      Update index.rst · 577dcd98
      Vittorio Caggiano authored
      fix max depth
      577dcd98
    • Vittorio Caggiano's avatar
      Update index.rst · eb2cabdc
      Vittorio Caggiano authored
      fix maxdepth
      eb2cabdc