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
In the current code, there is a lack of processing for offload when comes to checkpoints.
For example, with offload, the device of optimizer state should be CPU, which leads to an error when comm.
Besides, when loading the checkpoint, the device of the optimizer state should follow the original one rather than the loaded one. However, now it's just the opposite.
Environment
No response
The text was updated successfully, but these errors were encountered:
🐛 Describe the bug
In the current code, there is a lack of processing for offload when comes to checkpoints.
For example, with offload, the device of optimizer state should be
CPU
, which leads to an error when comm.Besides, when loading the checkpoint, the device of the optimizer state should follow the original one rather than the loaded one. However, now it's just the opposite.
Environment
No response
The text was updated successfully, but these errors were encountered: