Use DS 3D reshaping utils - SemanticDiff

Commit

3 years ago

Use DS 3D reshaping utils

References

#239 - Reshape deepspeed checkpoint

#289 - No-ZeRO reshaping

#292 - a branch combining layer-norm-auto-sync and ds_ckpt_reshape

Author

tjruwase

tjruwase

Parents

Loading