记录下这版动作 lora 的训练参数,MK5 翻车 MK6 重开中

MK6 训练日志,接着上一版往下记。

MK5 基本判定失败,直接废了。MK6 重新开练,计划 12000 步,现在才跑到第 1000 步,但出来的效果已经比之前所有版本都强不少,看着有戏。所以先把参数记下来,免得回头自己都忘了改过哪些。

数据集没动,沿用 MK5 那 2500 组配对,768×768、121 帧,算下来素材一共 7500 个文件。数据不变,前后两版的差别就基本只在参数上,方便对比。

这次改了这几处:

learning_rate: 0.0001 -> 0.0002
scheduler: linear
optimizer: AdamW
steps: 8000 -> 12000
batch: 2
first_frame: 0.4 -> 0.5
downscale: 2

整轮在一张 rtx 6000 pro 上跑,预计要 45 小时以上,只能先挂着,等跑完再回来贴对比。

心里没底的是学习率翻倍、步数又拉长,两个叠在一起后期会不会过拟。有经验的帮忙看看这套配置合不合理,给点意见就行。

MK5 是哪里崩的?过拟合还是动作压根学不进去,想避个坑

同问,MK5的loss曲线要是能贴一下就好了

1 个赞

我猜是数据里动作姿态太单一,之前也是这样,换个角度就学不会