- 17 Jul, 2021 1 commit
-
-
X Wang authored
* local_rank and install cuda version fix
-
- 02 Dec, 2020 1 commit
-
-
Janusz Lisiecki authored
- resume() is a nested function and when it loads best_prec1 it creates a local variable that hides the one from the parent function (which refers to the global one). This PR adds `global` to modify the global variable as intended Signed-off-by:
Janusz Lisiecki <jlisiecki@nvidia.com>
-
- 29 Sep, 2020 1 commit
-
-
ptrblck authored
-
- 28 Jan, 2020 1 commit
-
-
Vitaly Fedyunin authored
-
- 27 Jul, 2019 1 commit
-
-
Syed Tousif Ahmed authored
-
- 22 Jun, 2019 1 commit
-
-
Michael Carilli authored
-
- 21 Jun, 2019 1 commit
-
-
Michael Carilli authored
-
- 18 Jun, 2019 1 commit
-
-
Michael Carilli authored
-
- 30 Apr, 2019 2 commits
-
-
ptrblck authored
* remove unused tens tensor in example/imagenet/main_amp.py * remove unused tens tensor in deprecated examples and tests/L1
-
Michael Carilli authored
-
- 15 Mar, 2019 1 commit
-
-
Michael Carilli authored
-
- 13 Mar, 2019 1 commit
-
-
Michael Carilli authored
-
- 09 Mar, 2019 1 commit
-
-
Michael Carilli authored
-
- 04 Mar, 2019 2 commits
-
-
Michael Carilli authored
-
Michael Carilli authored
-
- 26 Feb, 2019 1 commit
-
-
Michael Carilli authored
-
- 02 Feb, 2019 1 commit
-
-
Michael Carilli authored
-
- 05 Jan, 2019 1 commit
-
-
Michael Carilli authored
-
- 04 Dec, 2018 1 commit
-
-
ptrblck authored
-
- 02 Dec, 2018 1 commit
-
-
ptrblck authored
-
- 01 Dec, 2018 1 commit
-
-
Michael Carilli authored
-
- 24 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 partia...
-
- 08 Oct, 2018 1 commit
-
-
Michael Carilli authored
-
- 14 Sep, 2018 1 commit
-
-
Michael Carilli authored
-
- 07 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
-
- 17 Aug, 2018 2 commits
-
-
Michael Carilli authored
-
Michael Carilli authored
-
- 23 Jul, 2018 1 commit
-
-
Michael Carilli authored
-
- 30 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.
-
- 16 Jun, 2018 1 commit
-
-
Michael Carilli authored
-
- 15 May, 2018 1 commit
-
-
Michael Carilli authored
-
- 06 May, 2018 1 commit
-
-
Michael Carilli authored
-
- 04 May, 2018 1 commit
-
-
Michael Carilli authored
-