如何修改XSeg默认的Batch_size大小

deepfacelabs 0 次点击

有时候我们在训练遮罩的时候发现默认的batch_size既然是4,这也太小了,影响遮罩训练速度。那么我们下面来修改一下

转到:“ DeepFaceLab_NVIDIA \ _internal \ DeepFaceLab \ models \ Model_XSeg”文件夹

用记事本打开“ Model.py”

找到以下行:

“ #override
def on_initialize_options(self):
self.set_batch_size(4)”

将“ 4”更改为任何值你喜欢。1060 GPU最大可以调至16,低于这个显卡的可以相应减少,自行测试一下。

保存“ Model.py”文件。