关于训练速度慢，CPU满载而显卡占用率低的一点解决方法

ZYDT

星级打分

• 1
• 2
• 3
• 4
• 5

平均分:NAN  参与人数:0  我的评分:未评

                                                                                                                                                        
                        

之前训练saehd模型，每运行几个迭代就会卡一次（单次时间变成14s左右），显卡几乎有一半的时间在偷懒，而CPU满载，打开网页都卡

后来去检查了下素材的分辨率，不知道自己什么时候抽风设的2000……

改回384之后，显卡满载，CPU占用率下降
方法如下：
分别运行2.31 data_src util faceset resize.bat和3.31 data_dst util faceset resize.bat
如果是猫之汉化版，我的方法是将_internal\.idea\inspectionProfiles\_script内的bat文件复制出来，然后打开(4.2) src头像 改变大小.bat和(5.2) dst头像 改变大小.bat

分辨率设到模型分辨率的两倍(不懂就设到320)，脸型直接按回车，提示是否合并，清选Y
然后运行，就会发现快很多啦

ccctttccct

终于知道卡的原因了，就是我的模型才192，我用1024去训练。卡出屎了

windjet

ZYDT 发表于 2023-6-4 20:16
也就是说这个resize是基于aligned内的图片进行的，而不是从帧图片进行重新提取？ ...

是的，这个resize是基于aligned内的图片进行的，所以landmark信息都会保留，不会丢掉以前的切脸信息。

windjet

ZYDT 发表于 2023-6-3 20:50
但是我听有人说素材分辨率应该设到模型分辨率的2倍，不知道什么原理

在素材与模型分辨率不一样的情况下，DFL都是先把素材resize到模型分辨率一样后再训练的，这个两倍的说法应该没有什么理论根据吧。
产生这种说法估计有两种可能，一是他觉得DFL的内部的resize处理的清晰度比他手上的resize图像处理工具要高，保留大一点的分辨率素材有助于提高训练清晰度，这在对比某些较差的工具的情况下的确有可能。二是素材不会总是变来变去的，考虑保留大一点的一份能对应更多及更高分辨率的模型。

windjet

之前的慢就是因为素材分辨率和模型分辨率不一样，所以DFL会在训练前先resize处理后再训练，resize处理就是速度慢的原因，你既然已经要把素材做resize处理了，为什么要把素材分辨率设到模型分辨率的两倍呢，设成和模型分辨率一样，不就免了DFL的resize处理会更快啊。

x777

嗯，谢谢分享

ZYDT

windjet 发表于 2023-6-3 18:56
之前的慢就是因为素材分辨率和模型分辨率不一样，所以DFL会在训练前先resize处理后再训练，resize处理就是 ...

但是我听有人说素材分辨率应该设到模型分辨率的2倍，不知道什么原理

ccctttccct

有可能，我以为呢。我跑1024也卡

ZYDT

windjet 发表于 2023-6-3 21:31
在素材与模型分辨率不一样的情况下，DFL都是先把素材resize到模型分辨率一样后再训练的，这个两倍的说法应 ...

如果换了高分辨率的模型，是不是可以用resize把分辨率调大呢？这样会不会损失质量？

windjet

ZYDT 发表于 2023-6-4 16:58
如果换了高分辨率的模型，是不是可以用resize把分辨率调大呢？这样会不会损失质量？ ...

用resize把分辨率调大不好，会损失清晰度质量的。

ZYDT

windjet 发表于 2023-6-4 17:52
用resize把分辨率调大不好，会损失清晰度质量的。

也就是说这个resize是基于aligned内的图片进行的，而不是从帧图片进行重新提取？