all gpu1<->2 reshapes work - SemanticDiff

Commit

3 years ago

all gpu1<->2 reshapes work

References

#239 - Reshape deepspeed checkpoint

#289 - No-ZeRO reshaping

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

Author

stas00

stas00

Parents

Loading