Skip to content

Handle actvitation checkpointing args that are None or non-tensors#660

Merged
ShadenSmith merged 2 commits intodeepspeedai:masterfrom
ShadenSmith:activation-ckpt-noneargs
Jan 12, 2021
Merged

Handle actvitation checkpointing args that are None or non-tensors#660
ShadenSmith merged 2 commits intodeepspeedai:masterfrom
ShadenSmith:activation-ckpt-noneargs

Conversation

@ShadenSmith
Copy link
Contributor

Fixes #598

@ShadenSmith ShadenSmith merged commit adcfd26 into deepspeedai:master Jan 12, 2021
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

bug Something isn't working

Projects

None yet

Development

Successfully merging this pull request may close these issues.

[deepspeed checkpointing] AttributeError: 'NoneType' object has no attribute 'numel'

2 participants