换头遮罩最重要的注意事项

别看我我不在

星级打分

• 1
• 2
• 3
• 4
• 5

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

    昨天兴起，画头部遮罩训练了许久。换头时依然只有半脸。感觉是遮罩有问题。在文件夹里查看，发现是最初始遮罩。
    大家记得要把模型文件夹里的遮罩删了，这样你只要画几张。训练遮罩时会让你新建一个遮罩。这时一定要选择head类型的。这时你就能得到一个换头遮罩了。
换头不仅需要用head类型的模式。遮罩，src,dst切脸都要是head类型。不用折腾，其它交给电脑。

wtxx8888

原版的训练遮罩 是固定且唯一的，固定位置 不存在文件，才能新建。
支持512遮罩的版本，可以多套遮罩并存，直接定个名字 新建一套完事。

别看我我不在

wtxx8888 发表于 2025-12-9 19:55
原版的训练遮罩 是固定且唯一的，固定位置 不存在文件，才能新建。
支持512遮罩的版本，可以多套遮罩并存， ...

我直接删了。我有多个文件夹。方便

别看我我不在

wtxx8888 发表于 2025-12-9 19:55
原版的训练遮罩 是固定且唯一的，固定位置 不存在文件，才能新建。
支持512遮罩的版本，可以多套遮罩并存， ...

大佬，头丹要多少代才能使用。我跑50万次左右了。最后一列还是很糊。清晰的少

wtxx8888

别看我我不在 发表于 2025-12-13 07:02
大佬，头丹要多少代才能使用。我跑50万次左右了。最后一列还是很糊。清晰的少 ...

糊就练呗，这东西 就没有定数。

别看我我不在

wtxx8888 发表于 2025-12-13 09:21
糊就练呗，这东西 就没有定数。

头丹果然 难练，100多万了效果没什么改善。就是源损失降到了0.4以下。看来可能要一个月才行

别看我我不在

wtxx8888 发表于 2025-12-9 19:55
原版的训练遮罩 是固定且唯一的，固定位置 不存在文件，才能新建。
支持512遮罩的版本，可以多套遮罩并存， ...

==                                                   ==
==            Model name: 256head-df-ud_SAEHD     ==
==                                                   ==
==     Current iteration: 1564531                    ==
==                                                   ==
==------------------ Model Options ------------------==
==                                                   ==
==            resolution: 256                        ==
==             face_type: head                       ==
==     models_opt_on_gpu: False                      ==
==                 archi: df-ud                      ==
==               ae_dims: 256                        ==
==                e_dims: 64                         ==
==                d_dims: 64                         ==
==           d_mask_dims: 22                         ==
==       masked_training: True                       ==
==       eyes_mouth_prio: False                      ==
==           uniform_yaw: False                      ==
==         blur_out_mask: True                       ==
==             adabelief: True                       ==
==            lr_dropout: y                          ==
==           random_warp: True                       ==
==      random_hsv_power: 0.0                        ==
==       true_face_power: 0.0                        ==
==      face_style_power: 0.0                        ==
==        bg_style_power: 0.0                        ==
==               ct_mode: none                       ==
==              clipgrad: False                      ==
==              pretrain: False                      ==
==       autobackup_hour: 3                          ==
== write_preview_history: False                      ==
==           target_iter: 0                          ==
==       random_src_flip: False                      ==
==       random_dst_flip: True                       ==
==            batch_size: 16                         ==
==             gan_power: 0.0                        ==
==        gan_patch_size: 64                         ==
==              gan_dims: 16                         ==
==                                                   ==
==------------------- Running On --------------------==
==                                                   ==
==          Device index: 0                          ==
==                  Name: NVIDIA GeForce RTX 4060 Ti ==
==                  VRAM: 13.25GB                    ==
==                                                   ==

大佬，请看我这参数有没有问题。批处理一开始是8，训练了很久没进步就改成了16，把模型和优化器放到了内存里。这思路是不是错的。
      要不我把批处理改成4，模型和优化器放回GPU试试。加快迭代是不是更重要？

wtxx8888

别看我我不在 发表于 2025-12-20 09:31
==                                                   ==
==            Model name: 256head-df-ud_S ...

16G显卡吧？256的BS，应该可以开 最少24吧？
开个毛毛的 lr_dropout: y？？？
models_opt_on_gpu: False？为什么不开？？？慢得要死！
而且，256的HEAD？只相当于WF的160.。。。下牙是出不来的！

别看我我不在

wtxx8888 发表于 2025-12-20 13:06
16G显卡吧？256的BS，应该可以开 最少24吧？
开个毛毛的 lr_dropout: y？？？
models_opt_on_gpu: False ...

我试过，最多开到8.多了就报错。是因为训练了许久没啥 进步，才加大到16，模型和优化器移到了内存上。
      不过，今天我把BS改成了4，先迭代到800万再说吧。我看到有人700万就相当 不错了。我现在284毫秒迭代一次。
模型和优化器移到内存里，批处理大小16时是890毫秒迭代一次。显存不够又想加大批处理只能如此。其实还是显卡在训练。

wtxx8888

别看我我不在 发表于 2025-12-20 17:46
我试过，最多开到8.多了就报错。是因为训练了许久没啥 进步，才加大到16，模型和优化器移到了内存上。
    ...

resolution: 256
ae_dims: 320
e_dims: 64
d_dims: 64
d_mask_dims: 32
这个四维，我1660T的6G显卡，models_opt_on_gpu: True 状态下，BS 6稳定 没任何问题。
云服务器上这个参数，models_opt_on_gpu: True 状态下，12G的3060 可以开BS 14。
你这四维 还低上一节的，models_opt_on_gpu: True 状态下，16G显卡开不到BS 16 ？？？