请问应该如何保存模型？

hpcaitech / ColossalAI-Examples

Examples of training models with hybrid parallelism using ColossalAI

Apache License 2.0

334 stars 102 forks source link

Closed 480284856 closed 2 years ago

480284856 commented 2 years ago

文档倒没问题欸，只是不知道如何保存训练好的模型。

binmakeswell commented 2 years ago

您好，可以参考这个https://www.colossalai.org/docs/basics/model_checkpoint/

480284856 commented 2 years ago

@binmakeswell 好的，谢谢！