DeepSpeed
tensorboard logging: avoid item() outside gas to improve performance
#5135

Merged

tensorboard logging: avoid item() outside gas to improve performance #5135

tjruwase merged 2 commits into deepspeedai:master from nelyahu:fix_tb_loss

tensorboard logging: avoid item() outside gas to improve performance

58a76e6f

nelyahu requested a review from

mrwyattii 2 years ago

nelyahu requested a review from

tjruwase 2 years ago

tjruwase approved these changes on 2024-02-15

Merge branch 'master' into fix_tb_loss

3b243b30

tjruwase merged fac1df99 into master 2 years ago

nelyahu deleted the fix_tb_loss branch 1 year ago

Reviewers

tjruwase

mrwyattii

Assignees

No one assigned

Labels

None yet

Milestone

No milestone