有时候我们在训练遮罩的时候发现默认的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”文件。