- 19 Feb, 2019 1 commit
-
-
Michael Carilli authored
-
- 01 Feb, 2019 1 commit
-
-
Michael Carilli authored
-
- 04 Dec, 2018 1 commit
-
-
ptrblck authored
-
- 02 Dec, 2018 1 commit
-
-
ptrblck authored
-
- 30 Nov, 2018 1 commit
-
-
Michael Carilli authored
-
- 23 Oct, 2018 1 commit
-
-
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 partially fixing launch configs enabling imagenet example to run with --sync_bn * [syncBN PR] Documentation added * adjusting README * adjusting again * added some doc to imagenet example * [syncBN] warp-level reduction bug fix: warp reduction logic updated. check for dummy element to avoid nan. improved launch config for better reduction kernels. Further improvements would be to increase grid size. * [syncBN] fixing undefined behavior in __shfl_down_sync from divergent threads in warp reduction. changing at::native::empty to at::empty (upstream comments)
-
- 07 Oct, 2018 1 commit
-
-
Michael Carilli authored
-
- 14 Sep, 2018 1 commit
-
-
Michael Carilli authored
-
- 06 Sep, 2018 1 commit
-
-
Michael Carilli authored
-
- 28 Aug, 2018 1 commit
-
-
Michael Carilli authored
-
- 20 Aug, 2018 1 commit
-
-
Michael Carilli authored
-
- 19 Aug, 2018 1 commit
-
-
Michael Carilli authored
-
- 16 Aug, 2018 2 commits
-
-
Michael Carilli authored
-
Michael Carilli authored
-
- 23 Jul, 2018 1 commit
-
-
Michael Carilli authored
-
- 29 Jun, 2018 2 commits
-
-
Michael Carilli authored
-
Josh Romero authored
Fixes to validation in imagenet example scripts. Precision and loss reporting modified to be consistent with train.
-
- 15 Jun, 2018 1 commit
-
-
Michael Carilli authored
-
- 14 May, 2018 1 commit
-
-
Michael Carilli authored
-
- 06 May, 2018 1 commit
-
-
Michael Carilli authored
-
- 03 May, 2018 1 commit
-
-
Michael Carilli authored
-