上周想给建筑线稿做个专门的控制分支,直接拿开源训练脚本改了改。最大的教训是数据对不齐——condition图和目标图必须像素级对应,我一开始随手resize,训出来全糊。
后来老老实实统一到512、把预处理固定死才好转。显存吃得也凶,batch开到4就爆,只能靠梯度累积凑。学习率我压到1e-5才稳,高了直接崩。
目前跑了几万步能看出控制力,但细节还差一截。有练过的佬吗,想请教下condition要不要做增强,怕加多了反而过拟合。
上周想给建筑线稿做个专门的控制分支,直接拿开源训练脚本改了改。最大的教训是数据对不齐——condition图和目标图必须像素级对应,我一开始随手resize,训出来全糊。
后来老老实实统一到512、把预处理固定死才好转。显存吃得也凶,batch开到4就爆,只能靠梯度累积凑。学习率我压到1e-5才稳,高了直接崩。
目前跑了几万步能看出控制力,但细节还差一截。有练过的佬吗,想请教下condition要不要做增强,怕加多了反而过拟合。