1. 17 Jul, 2021 1 commit
  2. 02 Dec, 2020 1 commit
  3. 29 Sep, 2020 1 commit
  4. 28 Jan, 2020 1 commit
  5. 27 Jul, 2019 1 commit
  6. 22 Jun, 2019 1 commit
  7. 21 Jun, 2019 1 commit
  8. 18 Jun, 2019 1 commit
  9. 30 Apr, 2019 2 commits
  10. 15 Mar, 2019 1 commit
  11. 13 Mar, 2019 1 commit
  12. 09 Mar, 2019 1 commit
  13. 04 Mar, 2019 2 commits
  14. 26 Feb, 2019 1 commit
  15. 02 Feb, 2019 1 commit
  16. 05 Jan, 2019 1 commit
  17. 04 Dec, 2018 1 commit
  18. 02 Dec, 2018 1 commit
  19. 01 Dec, 2018 1 commit
  20. 24 Oct, 2018 1 commit
    • jjsjann123's avatar
      [syncBN] (#48) · 81eef1ef
      jjsjann123 authored
      * [syncBN]
        added syncBN in native pure python apex
        added fused cuda kernels used for sync BN. Using welford for mean/var
          optional installation using 'python setup.py install --cuda_ext'
        added unit test with side to side comparison between apex sync BN with
          PyTorch BN. Notice that for pytorch BN implementation, because of
          numerical issue for mean/var, the output will be slightly off.
      
      * [syncBN PR]
        added fp16 support
        addressing review comments on:
          1. updating last pow 2
          2. look for import error when importing syncBN kernel
      
      * [syncBN PR]
        added convert function to insert SyncBatchNorm
        refactored some kernel code
      
      * fixing type issue (fp16/fp32/fp64)
      added Kahan summation
      editing unit test to use pytorch primitive ops with double, passing reasonable tests now
      
      * updating tensor creation calls
      
      * fixing the all_reduce contiguous tensor
      
      * transposed all reduce results
      
      * [syncBN]
      support fp16 input & fp32 layer for apex fp16
      partia...
      81eef1ef
  21. 08 Oct, 2018 1 commit
  22. 14 Sep, 2018 1 commit
  23. 07 Sep, 2018 1 commit
  24. 28 Aug, 2018 1 commit
  25. 20 Aug, 2018 1 commit
  26. 19 Aug, 2018 1 commit
  27. 17 Aug, 2018 2 commits
  28. 23 Jul, 2018 1 commit
  29. 30 Jun, 2018 2 commits
  30. 16 Jun, 2018 1 commit
  31. 15 May, 2018 1 commit
  32. 06 May, 2018 1 commit
  33. 04 May, 2018 1 commit