【Bug已解决】KeyError in optimizer.state_dict() under FSDP2 when using Adagrad optimizer 解决方案
【Bug已解决】KeyError in optimizer.state_dict() under FSDP2 when using Adagrad optimizer 解决方案
一、现象长什么样
用 FSDP2 训练,优化器是 Adagrad,在保存 checkpoint 调用 optimizer.state_dict()(或 accelerator.save_state&#…
2026/8/2 18:34:06