You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Describe the bug
Checkpointing crashes when --zero is set, with the error RuntimeError: Tensors must be CUDA and dense being thrown inside the method consolidate_state_dict()
Expected behavior
Shouldn't crash
Screenshots
The text was updated successfully, but these errors were encountered:
Describe the bug
Checkpointing crashes when
--zero
is set, with the errorRuntimeError: Tensors must be CUDA and dense
being thrown inside the methodconsolidate_state_dict()
Expected behavior
Shouldn't crash
Screenshots
The text was updated successfully, but these errors were encountered: