1. 30 Nov, 2020 1 commit
  2. 27 Nov, 2020 1 commit
  3. 26 Nov, 2020 1 commit
  4. 24 Nov, 2020 1 commit
  5. 22 Nov, 2020 1 commit
  6. 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
  7. 20 Nov, 2020 1 commit
  8. 19 Nov, 2020 4 commits
  9. 18 Nov, 2020 2 commits
  10. 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
  11. 16 Nov, 2020 1 commit
  12. 12 Nov, 2020 2 commits
  13. 11 Nov, 2020 2 commits
  14. 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
  15. 06 Nov, 2020 2 commits
  16. 04 Nov, 2020 1 commit
  17. 30 Oct, 2020 2 commits
  18. 29 Oct, 2020 1 commit
  19. 28 Oct, 2020 2 commits
  20. 26 Oct, 2020 1 commit
  21. 23 Oct, 2020 3 commits
  22. 22 Oct, 2020 3 commits
  23. 21 Oct, 2020 5 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
    • Vittorio Caggiano's avatar
      Update index.rst · 16b50272
      Vittorio Caggiano authored
      16b50272