luxiaolili / IDM-VTON-train

53 stars 3 forks source link

大佬求求求求求回复0.0 #9

Open TnoobT opened 2 months ago

TnoobT commented 2 months ago

大佬请教一下训练的一些细节:我用官方的idm代码训练,训练效果不稳定,有时候效果很差(mask都没inpaint完),有时候效果一般,每个epoch都在波动。请问这正常吗,是idm训练代码有问题,还是说其他的inpaint模型也是这样的。loss也是一直都在0.02-0.05波动,收敛不了。如果我用您的代码训练,效果能复现论文的结果吗,有什么需要注意的吗,是要每个checkpoint都保存吗。 ![Uploading image.png…]()

TnoobT commented 2 months ago

WX20240918-105821

luxiaolili commented 2 months ago

inpaint的模型训练不是很稳定,确实每个epoch都在波动,我batch为1训练是5000steps保存一次,一般从2万steps起开始测试,训练5-6万steps测一个效果好的模型,用我的代码能跑出来我给的例子,用测试集跑还可以,和作者的论文的结果我没有具体比对

TnoobT commented 2 months ago

inpaint的模型训练不是很稳定,确实每个epoch都在波动,我batch为1训练是5000steps保存一次,一般从2万steps起开始测试,训练5-6万steps测一个效果好的模型,用我的代码能跑出来我给的例子,用测试集跑还可以,和作者的论文的结果我没有具体比对

好的,感谢作者的回复。我看您好像没用ipadapter的权重好像

TnoobT commented 2 months ago

inpaint的模型训练不是很稳定,确实每个epoch都在波动,我batch为1训练是5000steps保存一次,一般从2万steps起开始测试,训练5-6万steps测一个效果好的模型,用我的代码能跑出来我给的例子,用测试集跑还可以,和作者的论文的结果我没有具体比对

其他的inpainting模型训练也是这样不稳定嘛

luxiaolili commented 2 months ago

inpaint的模型训练不是很稳定,确实每个epoch都在波动,我batch为1训练是5000steps保存一次,一般从2万steps起开始测试,训练5-6万steps测一个效果好的模型,用我的代码能跑出来我给的例子,用测试集跑还可以,和作者的论文的结果我没有具体比对

好的,感谢作者的回复。我看您好像没用ipadapter的权重好像 官方的ipadapter是训练人像,衣服不适用肯定是重新训练的加不加载预训练测试下来没关系

luxiaolili commented 2 months ago

inpaint的模型训练不是很稳定,确实每个epoch都在波动,我batch为1训练是5000steps保存一次,一般从2万steps起开始测试,训练5-6万steps测一个效果好的模型,用我的代码能跑出来我给的例子,用测试集跑还可以,和作者的论文的结果我没有具体比对

其他的inpainting模型训练也是这样不稳定嘛

只是训练了换衣服的inpainting模型其他的inpainting模型不太清楚。这个其实和官方提供的inpainting是一样的,不过它多加了一些condition的channel和把encoder换成ipadapter训练难度应该还好,训练了SVD模型,那个才是训练困难还很不稳定

TnoobT commented 2 months ago

inpaint的模型训练不是很稳定,确实每个epoch都在波动,我batch为1训练是5000steps保存一次,一般从2万steps起开始测试,训练5-6万steps测一个效果好的模型,用我的代码能跑出来我给的例子,用测试集跑还可以,和作者的论文的结果我没有具体比对

其他的inpainting模型训练也是这样不稳定嘛

只是训练了换衣服的inpainting模型其他的inpainting模型不太清楚。这个其实和官方提供的inpainting是一样的,不过它多加了一些condition的channel和把encoder换成ipadapter训练难度应该还好,训练了SVD模型,那个才是训练困难还很不稳定

哦哦,好的,谢谢大佬。能加个v吗0.0

TnoobT commented 2 months ago

inpaint的模型训练不是很稳定,确实每个epoch都在波动,我batch为1训练是5000steps保存一次,一般从2万steps起开始测试,训练5-6万steps测一个效果好的模型,用我的代码能跑出来我给的例子,用测试集跑还可以,和作者的论文的结果我没有具体比对

其他的inpainting模型训练也是这样不稳定嘛

只是训练了换衣服的inpainting模型其他的inpainting模型不太清楚。这个其实和官方提供的inpainting是一样的,不过它多加了一些condition的channel和把encoder换成ipadapter训练难度应该还好,训练了SVD模型,那个才是训练困难还很不稳定

大佬,我可以花钱请教你吗,求加v指导

luxiaolili commented 2 months ago

inpaint的模型训练不是很稳定,确实每个epoch都在波动,我batch为1训练是5000steps保存一次,一般从2万steps起开始测试,训练5-6万steps测一个效果好的模型,用我的代码能跑出来我给的例子,用测试集跑还可以,和作者的论文的结果我没有具体比对

其他的inpainting模型训练也是这样不稳定嘛

只是训练了换衣服的inpainting模型其他的inpainting模型不太清楚。这个其实和官方提供的inpainting是一样的,不过它多加了一些condition的channel和把encoder换成ipadapter训练难度应该还好,训练了SVD模型,那个才是训练困难还很不稳定

大佬,我可以花钱请教你吗,求加v指导

inpaint的模型训练不是很稳定,确实每个epoch都在波动,我batch为1训练是5000steps保存一次,一般从2万steps起开始测试,训练5-6万steps测一个效果好的模型,用我的代码能跑出来我给的例子,用测试集跑还可以,和作者的论文的结果我没有具体比对

其他的inpainting模型训练也是这样不稳定嘛

只是训练了换衣服的inpainting模型其他的inpainting模型不太清楚。这个其实和官方提供的inpainting是一样的,不过它多加了一些condition的channel和把encoder换成ipadapter训练难度应该还好,训练了SVD模型,那个才是训练困难还很不稳定

大佬,我可以花钱请教你吗,求加v指导

那倒不至于简单的可以帮忙,我在忙工作,可以加vx:wq1443961545