How to load_checkpoint using origin Megatron while used dlrover save_checkpoint ?

[LDH007]

Use DLRover's save_checkpoint function, result format like
iter_0000030/mp_rank_00_000/
iter_0000030/mp_rank_00_001/
iter_0000030/mp_rank_01_000/
...

But when I load_checkpoint using Megatron-LM's load_checkpoint It raises
[Errno 2] No such file or directory: 'iter_0000030/mp_rank_00/model_optim_rng.pt'
could not load the checkpoint

How to Solve This ?

[workingloong]

You use dlrover's load_checkpoint and Megatron-LM's save_checkpoint to transform the checkpoint format.