欢迎光临散文网 会员登陆 & 注册

【LoRA训练】不用再手动填写学习率,AI教你怎么训练自己。使用其他的Optimizer。

2023-02-23 21:44 作者:SHIN_NA  | 我要投稿

    kohya的训练程序更新了adafactor,Dadaption,LION等新的optimizer。其中有一些诸如adafactor的optimizer不需要填写学习率。同时效果甚至比人工经验的学习率训练出的模型效果更好。

    这代表以后可以不填写学习率让训练程序自己动了。

    首先更新kohya的训练程序。参数进行如下设置:

        其他随便填,Learning rate直接填写1,Optimizer选择Adafactor(据GUI作者说这个效果最好)。直接点击训练。

Learning rate直接填写1就好了,其他随意
训练脚本会有提示

等待训练完成就行,十分的傻瓜式。

    效果如下:

v2是之前的参数,似乎被薄纱了,奇树可爱

    loss以及lr参考:

学习率怎么在增加啊

    还是AI更懂AI。

【LoRA训练】不用再手动填写学习率,AI教你怎么训练自己。使用其他的Optimizer。的评论 (共 条)

分享到微博请遵守国家法律