聊聊训LoRA时epoch到底怎么定,别再无脑拉高了

epoch不是越高越好。训LoRA一上来就拉到几十,出图发僵、只会复刻训练集那几个角度,多半就是过拟合了。

epoch说白了就是整套训练图完整过几遍。图少、遍数又多,模型就把细节死记硬背下来,换个角度换个动作就不会了,也就是泛化差。

我自己的习惯:先给个不高的遍数跑一版,中途多存几个存档,挨个拿去出图试,挑那个既像、又还能自由发挥一点的。不认准最后一版,最后一版不一定是最好用的。

还有,步数是epoch乘图片数量、再跟batch有关,只盯一个数字看没意义。图片质量和打标不过关的话,遍数堆再多也白搭。

存档挑图这招太实用了,我以前只用最后一版

过拟合+1