DeepSpeed
[Stage][Fix] Add additional conditions when checking types of output from the model
#1026

Merged

[Stage][Fix] Add additional conditions when checking types of output from the model #1026

tjruwase merged 4 commits into deepspeedai:master from fix/hook_types

Add additional conditions when checking types of output from the model

8da9b19f

SeanNaren requested a review from

arashashari 4 years ago

SeanNaren requested a review from

awan-10 4 years ago

SeanNaren requested a review from

cli99 4 years ago

SeanNaren requested a review from

conglongli 4 years ago

SeanNaren requested a review from

eltonzheng 4 years ago

SeanNaren requested a review from

jeffra 4 years ago

SeanNaren requested a review from

minjiaz 4 years ago

SeanNaren requested a review from

niumanar 4 years ago

SeanNaren requested a review from

RezaYazdaniAminabadi 4 years ago

SeanNaren requested a review from

samyam 4 years ago

SeanNaren requested a review from

ShadenSmith 4 years ago

SeanNaren requested a review from

tjruwase 4 years ago

Add test

cdb3245b

tjruwase commented on 2021-04-30

Merge branch 'master' into fix/hook_types

2005bf76

SeanNaren changed the title ~~[Stage 3 Fix] Add additional conditions when checking types of output from the model~~ [Stage][Fix] Add additional conditions when checking types of output from the model 4 years ago

Modify test to use torch.tensor as well

2e8f09eb

tjruwase approved these changes on 2021-05-01

tjruwase merged b3870363 into master 4 years ago

Reviewers

tjruwase

arashashari

awan-10

cli99

conglongli

eltonzheng

jeffra

minjiaz

niumanar

RezaYazdaniAminabadi

samyam

ShadenSmith

Assignees

No one assigned

Labels

None yet

Milestone

No milestone

DeepSpeed [Stage][Fix] Add additional conditions when checking types of output from the model #1026 Merged

[Stage][Fix] Add additional conditions when checking types of output from the model #1026

DeepSpeed
[Stage][Fix] Add additional conditions when checking types of output from the model
#1026

Merged