DeepSpeed
fac1df99
- tensorboard logging: avoid item() outside gas to improve performance (#5135)
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Hide Minimap (CTRL+M)
Commit
1 year ago
tensorboard logging: avoid item() outside gas to improve performance (#5135) item() causes device/host synchronization Co-authored-by: Olatunji Ruwase <olruwase@microsoft.com>
References
#5135 - tensorboard logging: avoid item() outside gas to improve performance
Author
nelyahu
Parents
2d0a6bc2
Files
1
deepspeed/runtime
engine.py
Loading