Akegarasu / dreambooth-autodl

dreambooth autodl 训练脚本
85 stars 17 forks source link

优化保存间隔防止填满临时盘 #2

Closed remiliacn closed 1 year ago

remiliacn commented 1 year ago

为什么

经过一定的测试,发现6+次储存可能会导致磁盘空间填满,如果磁盘空间被填满会导致转化ckpt以及训练出错

解决方法

使用计算保证最多只会保存6次(5 + 1次last checkpoint)

例子:

测试

已进行训练测试。

Akegarasu commented 1 year ago

考虑到无代码基础用户需要手动更改这个参数,请优先添加一个新的 bool 参数如 auto_save_interval,启用后才更改已有保存间隔

remiliacn commented 1 year ago

加了一个参数using_auto_save_interval,如果为True则使用自动保存间隔,反则用用户设置。