DeepSpeed
Use amp autocast in ZeRO3 linear
#990
Merged

Commits
  • Use amp autocast in ZeRO3 linear
    tjruwase committed 5 years ago
  • Merge branch 'master' into olruwase/zero3_amp_autocast
    tjruwase committed 5 years ago
  • Fix typo
    tjruwase committed 5 years ago
  • Merge branch 'olruwase/zero3_amp_autocast' of github.com:microsoft/DeepSpeed into olruwase/zero3_amp_autocast
    tjruwase committed 5 years ago
  • Handle specific exceptions
    tjruwase committed 5 years ago
  • CI breaks on torch.distributed
    tjruwase committed 5 years ago
  • Merge branch 'master' into olruwase/zero3_amp_autocast
    tjruwase committed 5 years ago
  • Merge branch 'master' into olruwase/zero3_amp_autocast
    tjruwase committed 5 years ago
  • Merge branch 'master' into olruwase/zero3_amp_autocast
    jeffra committed 5 years ago
  • Add autocast unit test
    tjruwase committed 5 years ago
  • Merge branch 'olruwase/zero3_amp_autocast' of github.com:microsoft/DeepSpeed into olruwase/zero3_amp_autocast
    tjruwase committed 5 years ago
  • Merge branch 'master' into olruwase/zero3_amp_autocast
    tjruwase committed 5 years ago
  • Format fixes
    tjruwase committed 5 years ago
  • Merge branch 'olruwase/zero3_amp_autocast' of github.com:microsoft/DeepSpeed into olruwase/zero3_amp_autocast
    tjruwase committed 5 years ago
  • Fix skip logic
    tjruwase committed 5 years ago
Loading