Error while loading the states of optimizer in Trainer - def load(self, path)

[OliviaWang123456]

Hi,
Thank you for your excellent code.

But when I use deepspeed ZeRO2 and resume training from the saved checkpoint, there is an Error

Where should I use the load function in the trainer to resume my training?

Thanks a lot.