1. 12 May, 2021 1 commit
    • anj-s's avatar
      [chore] Rename and move checkpoint_activations from misc folder. (#654) · 72c6bab2
      anj-s authored
      * rename files
      
      * add newly renamed file
      
      * rename and move checkpoint activations related files
      
      * add test files to ci list
      
      * fix lint errors
      
      * modify docs
      
      * add changelog
      
      * retain old path for now
      
      * fix lint errors
      
      * add another import test case
      
      * fix merge conflict
      
      * add missing test file
      72c6bab2
  2. 15 Apr, 2021 1 commit
  3. 02 Apr, 2021 2 commits
  4. 08 Mar, 2021 1 commit
  5. 23 Feb, 2021 1 commit
  6. 19 Feb, 2021 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. 18 Nov, 2020 1 commit
  9. 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
  10. 12 Sep, 2020 1 commit