您好,我看您的 train.py 文件中没有加入 MegatronHook,需要有这个 hook 才

您好,我看您的 train.py 文件中没有加入 MegatronHook,需要有这个 hook 才能正常保存 checkpoint~可以参考 https://modelscope.cn/models/damo/nlp_gpt3_text-generation_1.3B/summary model card 中的训练代码示例 8014db2d86418a4ae8b01fca2b30422f.png

展开
收起
真的很搞笑 2023-04-22 12:48:19 100 分享 版权
1 条回答
写回答
取消 提交回答
  • 您好,感谢您的建议。我将尽快在 train.py 文件中加入 MegatronHook。

    2024-02-28 17:35:28
    赞同 展开评论

ModelScope旨在打造下一代开源的模型即服务共享平台,为泛AI开发者提供灵活、易用、低成本的一站式模型服务产品,让模型应用更简单!欢迎加入技术交流群:微信公众号:魔搭ModelScope社区,钉钉群号:44837352

热门讨论

热门文章

还有其他疑问?
咨询AI助理