Commits · 82d7a3bf0555004c653fac877410236dcbaa2b7a · OpenDAS / apex

15 Jun, 2018 3 commits
- Updating READMEs and examples · 82d7a3bf
  Michael Carilli authored Jun 15, 2018
  
  82d7a3bf
- Merge branch 'master' of https://github.com/NVIDIA/apex · 2c175a5d
  Michael Carilli authored Jun 14, 2018
  
  2c175a5d
- Updating READMEs · e215dd41
  Michael Carilli authored Jun 14, 2018
  
  e215dd41
14 Jun, 2018 10 commits
- Update README.md · 38cb0ce4
  mcarilli authored Jun 14, 2018
  
  38cb0ce4
- Merge branch 'master' of https://github.com/NVIDIA/apex · 18afea3b
  Michael Carilli authored Jun 14, 2018
  
  18afea3b
- Adding closure and minimal examples · f5790a1e
  Michael Carilli authored Jun 14, 2018
  
  f5790a1e
- Update README.md · 671f976e
  mcarilli authored Jun 14, 2018
  
  671f976e
- Merge branch 'master' of https://github.com/NVIDIA/apex · 71671993
  Michael Carilli authored Jun 14, 2018
  
  71671993
- Editing Dockerfile · b78b5ea5
  Michael Carilli authored Jun 14, 2018
  
  b78b5ea5
- Pulling in old logic to manually look for CUDA_HOME in Pytorch <= 0.4 to allow cross-compilation · 98fa5a3b
  Michael Carilli authored Jun 14, 2018
  
  98fa5a3b
- Added Dockerfile example, more readme updates · 0f703d13
  Michael Carilli authored Jun 14, 2018
  
  0f703d13
- Changing loss_scale to static_loss_scale in imagenet/main.py to be explicit · 942174bf
  Michael Carilli authored Jun 14, 2018
  
  942174bf
- Rewiring READMEs · 2a2341c7
  Michael Carilli authored Jun 14, 2018
  
  2a2341c7
13 Jun, 2018 2 commits
- Merge pull request #11 from NVIDIA/amp_lstm_backward · 43d1ae08
  Carl Case authored Jun 12, 2018
```
Handle the use of .sum() in fused LSTM/GRU backward
```
  43d1ae08
- Handle the use of .sum() in fused LSTM/GRU backward · 32fbbe48
  Carl Case authored Jun 12, 2018
  
  32fbbe48
12 Jun, 2018 2 commits
- Merge pull request #10 from NVIDIA/amp_cpu_fix · 227a9a2d
  Carl Case authored Jun 12, 2018
```
amp shouldn't convert CPU tensors
```
  227a9a2d
- Don't touch gpu tensors in functional API · f32a0a63
  Carl Case authored Jun 12, 2018
  
  f32a0a63
11 Jun, 2018 4 commits
- Merge pull request #9 from NVIDIA/imagenet_fix · 378ce1e1
  mcarilli authored Jun 11, 2018
```
DDP fix, imagenet speedup
```
  378ce1e1
- [Imagenet example] Switch validation to same I/O pipeline so validation produces correct result. · 06ee98c2
  Christian Sarofeen authored Jun 08, 2018
  
  06ee98c2
- Switched to non-bucketed comm · 421c9e66
  Christian Sarofeen authored Jun 08, 2018
  
  421c9e66
- Better shortcut for shared_param = True · 137d822b
  Christian Sarofeen authored Jun 08, 2018
  
  137d822b
08 Jun, 2018 1 commit
- Compilation succeeds on 0.4, 18.04-6 containers, and current upstream master · fb075b86
  Michael Carilli authored Jun 08, 2018
  
  fb075b86
07 Jun, 2018 2 commits
- Reverting setup.py to rely on cpp_extension's native CUDA_HOME, because... · bf855389
  Michael Carilli authored Jun 07, 2018
```
Reverting setup.py to rely on cpp_extension's native CUDA_HOME, because upstream fixed cross-compilation
```
  bf855389
- Updating setup.py to enable cross-compilation · 70b33770
  Michael Carilli authored Jun 06, 2018
  
  70b33770
06 Jun, 2018 4 commits
- Merge changes from Christian's fork · af431d28
  Michael Carilli authored Jun 06, 2018
  
  af431d28
- Updating latest amp changes to use new C++ backend · d6db91a4
  Michael Carilli authored Jun 06, 2018
  
  d6db91a4
- Update README · 89564b69
  Michael Carilli authored Jun 06, 2018
  
  89564b69
- Macros based on torch.__version__ to compile with 0.4 and 0.5 · d506eff2
  Michael Carilli authored Jun 06, 2018
  
  d506eff2
05 Jun, 2018 2 commits
- Merging latest master · 61b452e8
  Michael Carilli authored Jun 05, 2018
  
  61b452e8
- remove cuda check in build_cffi to handle docker installs · db6ae13a
  Carl Case authored Jun 05, 2018
  
  db6ae13a
01 Jun, 2018 3 commits
- Merge pull request #7 from pooyadavoodi/world_size_multiproc · 6143b30f
  Christian Sarofeen authored Jun 01, 2018
```
Read world_size argument in multiproc if specified in argslist
```
  6143b30f
- Merge pull request #8 from pooyadavoodi/speed_metric · eb0e59eb
  Christian Sarofeen authored Jun 01, 2018
```
Print speed (img/sec) in imagenet example
```
  eb0e59eb
- Print speed (img/sec) in imagenet example · dcda3b56
  Pooya Davoodi authored Jun 01, 2018
  
  dcda3b56
31 May, 2018 1 commit
- Read world_size argument in multiproc if specified in argslist · 38ea0e90
  Pooya Davoodi authored May 31, 2018
  
  38ea0e90
30 May, 2018 6 commits
- readme · 5ae9fbc4
  Carl Case authored May 30, 2018
  
  5ae9fbc4
- Hard ban on fp16 BCELoss · fb419005
  Carl Case authored May 30, 2018
  
  fb419005
- rename build->init · 6d30e1ff
  Carl Case authored May 30, 2018
  
  6d30e1ff
- WIP: better annotation / user function registry support · ec431d33
  Carl Case authored May 30, 2018
  
  ec431d33
- support multi-loss scaling per-optimizer correctly · 614b11ff
  Carl Case authored May 29, 2018
  
  614b11ff
- bugfix: keep cache up-to-date on parameter require_grad-ness · 8be1b053
  Carl Case authored May 29, 2018
  
  8be1b053