Open-Sora icon indicating copy to clipboard operation
Open-Sora copied to clipboard

建议train脚本在保存模型权重时加save_total_limit功能,如果训练数据集、训练epoch较大,不对保存的权重数量加以限制,你的磁盘会很快被打满,除非磁盘容量很大,感谢你们的工作,但是代码仍需严谨和规范

Open FrankMinions opened this issue 1 year ago • 4 comments

FrankMinions avatar Jul 11 '24 11:07 FrankMinions

我fork了你们的工作,并加入相关代码,如果需要快速使用的可以参考:https://github.com/FrankMinions/Open-Sora/blob/main/scripts/train.py

FrankMinions avatar Jul 11 '24 11:07 FrankMinions

I have encountered the same problem. Thanks for generously share your code.

CIntellifusion avatar Jul 13 '24 03:07 CIntellifusion

Thanks. Can you help make a PR if you want?

JThh avatar Jul 17 '24 00:07 JThh

This issue is stale because it has been open for 7 days with no activity.

github-actions[bot] avatar Jul 24 '24 01:07 github-actions[bot]

This issue was closed because it has been inactive for 7 days since being marked as stale.

github-actions[bot] avatar Sep 12 '24 01:09 github-actions[bot]