可以包含 DeepNVMe + 状态字典的示例

作者: sayakpaul创建于 2025年6月19日更新于 2025年6月20日

https://GitHub.com/deepspeedai/DeepSpeedExamples/tree/master/deepnvme/file_access 中的示例非常好。但是,问题在于现实世界的应用需要能够将模型状态字典传递给方法并加载它们。是否有必要添加此类工具?还是建议将其与 https://GitHub.com/deepspeedai/DeepSpeedExamples/tree/master/deepnvme/model_checkpoint/torch 结合使用(这有点困难,因为需要修改现有的 torch 安装)?

内容来源: deepspeedai/DeepSpeedExamples