DeepSpeed
57140e8e - fix: fix BF16_Optimizer compatibility issue with optimizer state 0-dim tensor (#2152)
Commit
View On GitHub
Comment changes are shownComment changes are hidden
Commit
2 years ago
fix: fix BF16_Optimizer compatibility issue with optimizer state 0-dim tensor (#2152) Co-authored-by: Olatunji Ruwase <olruwase@microsoft.com>
Author
shjwudp shjwudp
Parents
66d29b0a
  • deepspeed/runtime
    • File
      bf16_optimizer.py
FAQ Terms Privacy Refunds Impressum