DeepSpeed
Use `torch.nan_to_num` replace numpy wrapper one
#5877

Merged

Use `torch.nan_to_num` replace numpy wrapper one #5877

loadams merged 4 commits into deepspeedai:master from jinyouzhi:replace_nan_to_num

Use `torch.nan_to_num` replace numpy wrapper one

b31907e8

jinyouzhi requested a review from

tjruwase 1 year ago

Merge branch 'master' into replace_nan_to_num

27426eec

add torch version detect

1120ca58

tjruwase approved these changes on 2024-08-13

loadams approved these changes on 2024-08-14

Merge branch 'master' into replace_nan_to_num

674b43a1

loadams enabled auto-merge 1 year ago

loadams merged 862aff37 into master 1 year ago

Reviewers

loadams

tjruwase

Assignees

No one assigned

Labels

None yet

Milestone

No milestone