Enhancement: Ability to load checkpoint without loading the optimizer… (#128)
* Enhancement: Ability to load checkpoint without loading the optimizer states. Unittest testing saving and loading checkpoint with fused, unfused and zero optimizer. The unitest takes about 165s
Showing
deepspeed/pt/deepspeed_zero_optimizer.py
100644 → 100755
deepspeed/pt/fp16_optimizer.py
100644 → 100755
tests/model/Megatron_GPT2/run_checkpoint_test.py
100644 → 100755
文件模式从 100644 更改为 100755
tests/unit/test_checkpointing.py
0 → 100755
想要评论请 注册 或 登录